ZOJ3747 Attack on Titans(容斥dp)

传送门

三个数字$1,2,3$组成长度$n$的序列

$0$的话无所谓

至少出现$m$个连续$1$

至多出现$k$个连续$2$

---

发现至少并不好求

转而求至多出现$k$个连续$2$,至多出现$m-1$个连续$1$,方案记作$an{s}_1$

然后求不限制$1$,至多出现$k$个连续$2$,方案数记住$ans2$

最终答案就是$ans2-ans1$

$T1$

我们先来求$ans2$,假如不考虑任何限制,定义$f[i][j]$为长$i$当前数字是$j$的方案数

$f[i][j]=f[i-1][0]+f[i-1][1]+f[i-1][2]$

现在需要至多出现$k$个连续$2$,那么如果在$i$位置填充$2$的话

如果$[i-k,i-1]$都是$2$那就不满足条件了,需要减去

$f[i][2]=sum[i-1]-f[i-k-1][0]-f[i-k-1][1]$

其中$sum[i]$表示长$i$的合法序列

$T2$

有了上面的方法,求$an{s}_1$也是一样了

当$i>m-1$时,当$[i-m+1,i]$都是$1$也不满足条件

$f[i][1]=sum[i-1]-f[i-m][0]-f[i-m][2]$

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int mod = 1e9+7;
const int maxn = 1000009;
int n,m,k;
int f[maxn][4];
signed main()
{
	while( cin >> n >> m >> k )
	{
		f[0][0] = 1;
		f[1][0] = f[1][1] = f[1][2] = 1;
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<=2;j++)	f[i][j] = ( f[i-1][0]+f[i-1][1]+f[i-1][2] )%mod;
			if( i>k )	f[i][2] = ( f[i][2]-f[i-k-1][0]-f[i-k-1][1] )%mod;
		}
		int ans2 = ( f[n][0]+f[n][1]+f[n][2] )%mod;
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<=2;j++)	f[i][j] = ( f[i-1][0]+f[i-1][1]+f[i-1][2] )%mod;
			if( i>k )	f[i][2] = ( f[i][2]-f[i-k-1][0]-f[i-k-1][1] )%mod;
			if( i>m-1 )	f[i][1] = ( f[i][1]-f[i-m][0]-f[i-m][2] )%mod;		
		}
		int ans1 = ( f[n][0]+f[n][1]+f[n][2] )%mod;
		cout << ( (ans2-ans1)%mod+mod )%mod << endl;
	}
}