「TJOI 2018」游园会

传送门

---

problem

给定一个模式串 $S$，长度为 $k$，只含字符 'N','O','I'。对于 $\forall i\in [0,k]$，求出满足以下条件的串的数量：

1. 长度为 $n$ 且只含字符 'N','O','I'。
2. 与 $S$ 的最长公共子序列长度恰好为 $i$。
3. 不含子串 NOI。

答案对 $10^9+7$ 取模。

数据范围：$n\le 1000$，$k\le 15$。

---

solution

参考博客。

首先考虑一下最长上升子序列怎么求，设 $d{p}_{i,j}$ 表示 $A$ 串前 $i$ 个字符，$B$ 串前 $j$ 个字符的答案，则：

l c s i , j = m a x { l c s i − 1 , j , l c s i , j − 1 , l c s i − 1 , j − 1 + [ A i = B j ] } \mathrm{lcs}_{i,j}=max\{\mathrm{lcs}_{i-1,j},\mathrm{lcs}_{i,j-1},\mathrm{lcs}_{i-1,j-1}+[A_i=B_j]\} lcsi,j​=max{lcsi−1,j​,lcsi,j−1​,lcsi−1,j−1​+[Ai​=Bj​]}

设 $B$ 串就是题中的模式串 $S$，是已知的。也就是说，我们要求 ${\mathrm{l}\mathrm{c}\mathrm{s}}_i$ 这一行，只需要 ${\mathrm{l}\mathrm{c}\mathrm{s}}_{i-1}$ 这一行和 $A_i$。

发现 ${\mathrm{l}\mathrm{c}\mathrm{s}}_{i,j}-{\mathrm{l}\mathrm{c}\mathrm{s}}_{i,j-1}\le 1$，那么对 ${\mathrm{l}\mathrm{c}\mathrm{s}}_i$ 差分一下会是一个长度为 $k$ 的 $01$ 串，是可以状态压缩的！

那么我们可以再设计一个 $dp$：$f_{i,j,0/1/2}$ 表示 $dp$ 到第 $i$ 位，${\mathrm{l}\mathrm{c}\mathrm{s}}_i$ 的状态为 $j$，第 $i$ 位填 'N'$/$'O'$/$'I' 的方案数。那么，我们可以把 ${\mathrm{l}\mathrm{i}\mathrm{s}}_i$ 还原成数组，按照 $\mathrm{l}\mathrm{i}\mathrm{s}$ 的 $dp$ 转移后得到新的状态，再对填 'N'$/$'O'$/$'I' 简单分类讨论一下得到转移方程。

注意在统计答案的时候，若状态 $i$ 中有 $num$ 个 $1$，那么 $i$ 的贡献应该加到 $num$ 里（因为 $i$ 是经过差分后再状压的）。

时间复杂度 $O(nk{2}^k)$。

---

code

#include<bits/stdc++.h>
using namespace std;
const int N=(1<<15)+5,P=1e9+7;
int n,k,f[2][N][3]={1},bit[N],ans[1005];
void Add(int &x,int y)  {x=(x+y>=P)?x+y-P:x+y;}
int A[20],B[20];
void Get_array(int sta){
	for(int i=0;i<k;++i)  A[i+1]=(sta>>i)&1;
	for(int i=1;i<=k;++i)  A[i]+=A[i-1];
}
int Get_sta(){
	int sta=0;
	for(int i=0;i<k;++i)  sta|=((B[i+1]-B[i])<<i);
	return sta;
}
char S[20];
void dp(int t,int j,int num,char c,int val){
	Get_array(j);
	for(int i=1;i<=k;++i)  B[i]=max(max(A[i],B[i-1]),A[i-1]+(c==S[i]));
	int x=Get_sta();
	Add(f[t][x][num],val);
}
int main(){
	scanf("%d%d%s",&n,&k,S+1);
	int sta=(1<<k),t=1;
	for(int i=1;i<=n;++i,t^=1){
		memset(f[t],0,sizeof(f[t]));
		for(int j=0;j<sta;++j){
			if(f[t^1][j][0]){
				dp(t,j,1,'N',f[t^1][j][0]),
				dp(t,j,0,'O',f[t^1][j][0]),
				dp(t,j,0,'I',f[t^1][j][0]);
			}
			if(f[t^1][j][1]){
				dp(t,j,1,'N',f[t^1][j][1]),
				dp(t,j,2,'O',f[t^1][j][1]),
				dp(t,j,0,'I',f[t^1][j][1]);
			}
			if(f[t^1][j][2]){
				dp(t,j,1,'N',f[t^1][j][2]),
				dp(t,j,0,'O',f[t^1][j][2]);
			}
		}
	}
	for(int i=0;i<sta;++i)  bit[i]=bit[i>>1]+(i&1);
	for(int i=0;i<sta;++i)
		for(int j=0;j<3;++j)  Add(ans[bit[i]],f[t^1][i][j]);
	for(int i=0;i<=k;++i)  printf("%d\n",ans[i]);
	return 0;
}