ZOJ 3725 Painting Storages DP

#include
#include
#include
using namespace std;
#define inf 1000000007
#define N 100005
int a[N];
int b[N];
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[n]);
	}
	return 0;
}
a[i]表示前i点满足条件的种类,对于当前点,这个点之前已经满足条件,则当前点随意,就是2种都行,如果之前还不能满足条件,如果要满足条件,则当前点一定是红色,第i-m那个点一定是蓝色(如果不为蓝色就满足条件了),对于剩下的所有点:一共有2^(i-1-m)中,再减去这里可能满足条件的情况就行了。

你可能感兴趣的:(dp)