FZU 2126 消去游戏 (dp)

题意：

开始在一个容器中给出n个颜色各异的球，当同种球连续k个时就会消掉。现在要求你继续在容器中放m个球，求能使容器的球变成0个的方案数。题目保证开始的n个球没有存在可以消去的情况。

题解：

dp[i][j]表示m个球中用了i个球，并且还需要j个球才能使得整个变成0。

dp[i][j]=dp[i-1][j-k+1]+dp[i-1][j+1]

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
typedef unsigned long long ll;
const int oo=0x3f3f3f3f;
const int MOD=1000000007;
const int maxn=1005;
int dp[maxn][maxn];

int DP(int n,int k,int h,int tol){

    memset(dp,0,sizeof dp);
    dp[0][tol]=1;
    for(int i=1;i<=n;i++){
        for(int j=n-i;j>=0;j--){
            if(j-k+1>=0)
                dp[i][j]=(dp[i][j]+(ll)dp[i-1][j-k+1]*(j-k+1==0?h:h-1)%MOD)%MOD;
            dp[i][j]=(dp[i][j]+dp[i-1][j+1])%MOD;
        }
    }
    return dp[n][0];
}

int main(){

    int n,m,k,h;
    while(scanf("%d %d %d %d",&n,&m,&h,&k)!=EOF){

        int pre=0,now=0,tol=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&now);
            if(now!=pre) tol+=k-1;
            else tol--;
            pre=now;
        }
        printf("%d\n",DP(m,k,h,tol));
    }
	return 0;
}