C语言动态规划(9)___消去游戏(FZU 2126)

Problem Description

S最近在玩一种游戏。这种游戏的规则是一个一个地往一个栈里放有颜色的球，当栈顶连续k个球颜色相同时，这k个球立刻同时消失。现在S已经往栈里放了n个球，他想知道再放m个球，然后使得栈里的球都被消去的放法有多少种。两种放法不同是指存在放的第i个球这两种放法放的球的颜色不同。由于方法数可能很多，将答案mod 1000000007。

 Input

输入包含多组数据。输入数据的第一行为四个整数n,m,h,k(0<=n,m,h<=1000,2<=k<=1000)，表示已经放了n个球，有h种不同颜色的球，若栈顶出现连续k个球颜色相同则这k个球同时消失，问再放m个球，使得最后栈里的球都被消去的放法数。第二行从左往右依次输入n个整数，范围为1到h,表示刚开始往栈里放的球的颜色，放入顺序与输入顺序相同，数据保证已经放入的n个球不会存在连续k个球颜色相同。答案对1000000007取余。

Output

输出一行一个整数M,表示对1000000007取余后的放法数。

Sample Input

3 6 3 3 1 2 20 6 2 3

Sample Output

98

分析：这道题如果用

搜索的话规模将达到 h ^ m ,最高可达到1000^1000 ;

那么我们尝试用动态规划.

这个时候,我们定义一个新的栈——虚拟栈.

那么我们定义一下虚拟栈的作用:

题目中给出了 n , m , h , k 四个变量.假如我们按照题目的测试数据来举例：

3 6 3 3

1 2 2

我们从题目的栈中来看：(接下来姑且称题目中的栈为实际栈)

实际栈先是1号颜色的球，那么我们就向虚拟栈中放 2个 ( k - 1个 )1号颜色的球。实际栈接下来是2号颜色的球，那么继续向虚拟栈中放2个2号颜色的球。实际栈栈顶还是2号颜色的球和实际栈前一个颜色相同，那么我们在虚拟栈中就减去一个2号颜色的球。处理完实际栈后我们可以发现虚拟栈中从先入开始分别是2个1号颜色，和一个2号颜色，共三个球。

到这里我们可以看出其实当前虚拟栈就是使得当前实际栈最快消完的方案。我们且忽略虚拟栈中各种球颜色的差别，那么虚拟栈中的球的个数就是当前实际栈要消完需要球的个数。

这时候我们可以定义一个二维数组 dp [  ] [  ];

那么 dp[ i ] [ j ] 中 i 表示还有i个球没放,j 表示虚拟栈中还有 j 个球。dp [ i ] [ j ] 的值表示到达这个状态的方法数.

那么我们可以知道：在dp [ i ] [ j ] ( i>0 && j>0 ) 时候，我们可以知道实际栈的球个数肯定不为0，假设实际栈栈顶的球颜色为s号.

这个时候如果我放一个s号球进去，那么虚拟栈的球个数少1，那么dp [ i ] [ j ] 就转化为 dp [ i-1 ] [ j-1 ],那么这种情况我们可以得出状态转移方程是 dp [ i-1 ] [ j-1 ] =( dp [ i ] [ j ] + dp [ i-1 ] [ j-1 ] );

如果我放的球不为s号，那么虚拟栈的球将会增加 k-1 个，那么dp [ i ] [ j ] 就转化为 dp [ i-1 ] [ j+k-1 ],那么这种情况我们可以知道.放入球的颜色有 h - 1 种 ( 特殊的,当实际栈没有球的时候,这时候放入球有 h 种 )。 所以状态转移方程式就是： dp[ i-1 ] [ j+k-1 ] = dp[ i-1 ] [ j+k-1 ]+dp [ i ] [ j ] *(j? h-1:h) ；

由于我们知道当 dp [ i ] [ j ] 中 j > i 的时候,表示的是虚拟栈的球的个数大于剩余放的球的个数，也就是说以当前虚拟栈最少的方案，也大于剩余球的个数，那么我们知道这种情况肯定是不能全部消完球的。所以以上状态转移方程都是在 i > j 这个条件下进行。

最后：dp [ 0 ] [ 0 ] 也就是我们最后要得到的答案。

代码：

#include<stdio.h>   
#include<string.h>  
#define mod 1000000007;  
int n,m,h,k;  
long long dp[1005][1005];
int main()
{
	int i,j,pre,cur,tot;
	int n,m,h,k;
	while(scanf("%d%d%d%d",&n,&m,&h,&k)!=EOF){
		tot=cur=pre=0;
		for(i=0;i<n;i++){
			scanf("%d",&cur);
			if(cur!=pre) tot+=k-1;   //实际栈当前颜色和上一个颜色不同时,虚拟栈放入k-1个该颜色球.
			else tot--;              //实际栈当前颜色和上一个颜色相同时,虚拟栈减去 1 个该颜色球.
			pre=cur;                //pre 和cur 分别是 实际栈上一个球和当前球的颜色.
		}
		if(tot>m){
			printf("0\n");
			continue;            //当虚拟栈的球数目大于剩余放球的个数,该情况无解.
		}
		memset(dp,0,sizeof(dp));
		dp[m][tot]=1;
		for(i=m;i>0;i--){
			for(j=i;j>=0;j--){
				if(i-1>=j+k-1) dp[i-1][j+k-1]=(dp[i-1][j+k-1]+dp[i][j]*(j? h-1:h))%mod;  
				if(j>0) dp[i-1][j-1]=(dp[i-1][j-1]+dp[i][j])%mod;
			}
		}
		printf("%lld\n",dp[0][0]);
	}
	return 0;
}