ZOJ 3725 Painting Storages DP

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define inf 1000000007
#define N 100005
int a
;
int b
;
void getsolve(){
b[0]=1;
for(int i=1;i<=N;i++){
b[i]=b[i-1]*2%inf;
}
return;
}
int main(){
int n,m;
getsolve();
while(scanf("%d%d",&n,&m)!=EOF){
memset(a,0,sizeof(a));
a[m]=1;
for(int i=m+1;i<=n;i++){
a[i]=(((a[i-1]*2)%inf+(b[i-1-m]-a[i-1-m])%inf)%inf+inf)%inf;
}
printf("%d\n",a
);
}
return 0;
}

a[i]表示前i点满足条件的种类，对于当前点，这个点之前已经满足条件，则当前点随意，就是2种都行，如果之前还不能满足条件，如果要满足条件，则当前点一定是红色，第i-m那个点一定是蓝色(如果不为蓝色就满足条件了)，对于剩下的所有点：一共有2^(i-1-m)中，再减去这里可能满足条件的情况就行了。