ARC 058E 和風いろはちゃん 状压DP(bit)

题意:长度为N,每个元素为1~10的某一个,总共有10^N种序列,现在给出N,X,Y,Z 问有多少个长度为N的序列满足其有四个下标x,y,z,w 

使得a[x]+a[x+1]..a[y-1]=X,a[y]+a[y+1]+..a[z-1]=Y,a[z]+a[z+1]+.a[w]=Z,

N<=40,X<=5,Y<=7,Z<=5

若找到四个下标,则剩余元素可以任意.
暴力枚举第一个x出现的位置和三段的长度即可确定一类解.因为其余元素任意 枚举时明显重复计算了..

考虑不包含有连续一段和为X,Y,Z的序列.
首先用0/1串代表数字 1="1",2="10",3="100",i.e:(1,2,1,4)="11011000",该段和为i.长度为i,第i位1

(X,Y,Z)=(5,7,5)禁止(X,Y,Z)状态为="10000100000010000",X+Y+Z=17则最多只需要用17个bit判断当前是否包含禁止状态,

设dp[i][sttae] 插最后一个数时更新即可.O(10*2^17*N)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1<<17;
const ll mod=1e9+7;
ll n,x,y,z;
ll dp[45][N];
int main()
{
	while(cin>>n>>x>>y>>z)
	{
		ll ans=1;
		for(int i=1;i<=n;i++)
			ans=(ans*10ll)%mod;
		memset(dp,0,sizeof(dp));
		dp[0][0]=1;
		int forbid=(1<<z-1)|(1<<y+z-1)|(1<<x+y+z-1);
		int state=(1<<x+y+z)-1;
		for(int i=1;i<=n;i++)
		{
			for(int s=0;s<=state;s++)
			{
				if(dp[i-1][s]==0)//
					continue;
				for(int k=1;k<=10;k++)
				{
					int now=(s<<k)|(1<<k-1);
					now&=state;
					if((now&forbid)!=forbid)
						dp[i][now]=(dp[i][now]+dp[i-1][s])%mod;
				}
			}
		}
		for(int i=0;i<=state;i++)
			ans=(ans-dp[n][i]+mod)%mod;
		cout<<ans<<endl;
	}
	return 0;
}