Codeforces 474 D Flowers（DP）

题目链接：

Codeforces 474 D Flowers

题意：

有两种花可以吃，white花只能连续吃k个，red花不受限制，当吃a到b朵花时一共有多少种吃法？也就是：

a朵花时的吃法+(a+1）朵花时的吃法+…+b朵花时的吃法。

分析：

用dp[i][0]和dp[i][1]分别表示吃第i朵花时是white和red的情况。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
using namespace std;
const int MAX_N=100010;
const int mod=(int)(1e9+7);

int T,k,a,b;
int dp[MAX_N][2],ans[MAX_N],sum[MAX_N]={0};

inline void solve()
{
for(int i=1;i<MAX_N;i++){
if(i<k) dp[i][0]=0,dp[i][1]=1;
else if(i==k) dp[i][0]=1,dp[i][1]=1;
else{
dp[i][0]=(dp[i-k][0]+dp[i-k][1])%mod; //ith eat white flower
dp[i][1]=(dp[i-1][0]+dp[i-1][1])%mod; //ith eat red flower
}
ans[i]=(dp[i][0]+dp[i][1])%mod; //ith total ways
sum[i]=(sum[i-1]+ans[i])%mod;   //from 1st to ith total ways
}
}

int main()
{
scanf("%d%d",&T,&k);
solve();
while(T--){
scanf("%d%d",&a,&b);
printf("%d\n",(sum[b]-sum[a-1]+mod)%mod); // +mod just in case of the result of sum[b]-sum[a-1] is negative
}
return 0;
}