Uva-1633 Dyslexic Gollum(状压DP)

题意:输入正整数n和k(1<=n<=400,1<=k<=10),求长度为n的01串中有多少个不含长度至少为k的回文连续子串.

分析:猛一看感觉没法做,这题的trick在,如果现在有一个超过k的连续回文子串存在,那么一定也存在一个小于k的回文子串,同理如果我们只要保证所有前缀中不存在超过k的回文子串,那么最后整个串一定也不存在超过k的回文子串,所以我们判断到k就行了(因为奇偶长度关系其实是k+1).k很小可以直接上状压dp,f[i][j]表示当前到第i位且后k+1位状态为j的方案数.

#include <bits/stdc++.h>
#define N 200005
#define MOD 1000000007
using namespace std;
int T,n,k,jud[1<<11];
long long ans,f[405][1<<11];
bool check(int x,int k)
{
	for(int i = 1;i <= k/2;i++)
	 if(((x & (1<<(i-1)))<<(k-2*i+1)) != (x & (1<<(k-i)))) return false;
	return true;
}
void Init()
{
	for(int i = 0;i < (1<<(k+1));i++) jud[i] = !(check(i,k) || check(i>>1,k) || check(i,k+1));		
}
int main()
{
	scanf("%d",&T);	
	while(T--)
	{
		scanf("%d%d",&n,&k);
		Init();
		memset(f,0,sizeof(f));
		ans = 0;
		f[1][0] = f[1][1] = 1;
		for(int i = 1;i <= n;i++)
		 for(int j = 0;j < (1<<min(i,k+1));j++)
		 {
		 	if(i == k && check(j,k)) continue;
		 	if(i > k && !jud[j]) continue;
		 	if(i != n)
		 	{
		 		if(i <= k)
		 		{
		 			f[i+1][j] = (f[i+1][j] + f[i][j]) % MOD;
					f[i+1][j+(1<<i)] = (f[i+1][j+(1<<i)] + f[i][j]) % MOD;	
				}
				else 
				{ 
					f[i+1][j>>1] = (f[i+1][j>>1] + f[i][j]) % MOD;
					f[i+1][(j>>1) + (1<<k)] = (f[i+1][(j>>1) + (1<<k)] + f[i][j]) % MOD;
				}
			}
			else ans = (ans + f[i][j]) % MOD;
		 }
	    cout<<ans<<endl;
	}
}