BZOJ3398 牡牛和牝牛 - 动态规划

传送门

题解：沙茶题。前缀和维护一下。

这题显然可以再O(k+lgn)的时间内出解。

代码：

#include
#include
#include
#define mod 5000011
#define N 100010
using namespace std;
int dp[N],s[N];
int main()
{
    int n,k;scanf("%d%d",&n,&k);
    dp[0]=s[0]=1;
    for(int i=1;i<=n;i++)
        dp[i]=((i-k-1>=0)?s[i-k-1]:1),
        s[i]=(s[i-1]+dp[i])%mod;
    printf("%d\n",s[n]);return 0;
}