禁止字符串——白书P368

题目：

考虑只由’A’, ‘G’, ‘C’, ‘T’四种字符组成的DNA字符串。给定一个长度为k的字符串S，请计算长度恰好为n且不包含S的字符串的个数。输出个数mod 10009后的结果。
其中 1<=k<=100 1<=n<=10000

样例

输入
n=3 k=2 S=“AT”

输出
56

题解

首先考虑最直观的算法就是：生成所有满足条件的字符串，但是字符串个数可能高达4^n，显然是行不通的。
接下来，与其在生成字符串后在判断它是否包含s，不如在搜索的过程中，每在末尾添加一个字符的时候，都确保其最后k个字符不包含s。可见最后k个字符之前的字符对以后的搜索并无影响。所以，我们可以以剩余字符的个数和最后k-1个字符为状态进行动态规划。

经过等价状态的归并后，其实只有k种状态。即已经生成的字符串的后缀和s的前缀的匹配长度作为状态。

思路：字符串预处理+动态规划。
在以下代码中有两个数组next[i][j] 与dp[i][j] 分别代表了以上思路的两个处理。

next[i][j] 表示的是当前枚举到的字符串的末尾的i个字符与S的前i个字符一样的时候，当下一个枚举到的字符是j（这里j的取值是0~4 分别代表着“A， G， C， T”）的时候，其构造的新串末尾有i个字符与S的的前i个字符一样的长度。

由以下例子可以更好地理解：

比如 S = “ATCG” next[1][0] 代表的是当前串的末尾是A现在多加一个A所以当前枚举到的串的末尾变成了AA， 与S相比其匹配数还是1；
所以： next[1][0] = 1;
同理 next[1][1] = 0;
对于S = “ATCG” next为：
1 0 0 0
1 0 0 2
1 0 3 0
1 4 0 0

有时会有重复的情况请按匹配最大值储存：例如 S = “ATCATCG”
如果当前枚举到的字符串末尾是ATCAT 下一位枚举的是C的话那么对于next[5][2] 可以等于3 也可以等于6， 这个时候按6算(按长的来算)。

dp[i][j] 表示前i个字符最后面j个字符与S的前j个字符匹配的合法的字符串的数目

状态转移方程：dp[i+1][j] += dp[i][m] ( 0 <= m <= k) ，在这里并非累加所有的，要去掉那些违反题目条件的。

代码：

#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
typedef long long ll;

int n,k;
string S;

const char *agct = "AGCT";

const int maxn = 1009;
int dp[maxn][108];
int nex[maxn][5];
const int MOD = 10009;
void init()
{
    memset(dp,0,sizeof(dp));
    dp[0][0] = 1;
}

void solve()
{
    for (int i=0;i<k;i++)
    {
        for (int j=0;j<4;j++)
        {
            string s = S.substr(0,i) + agct[j];// 取S的前i个元素并在后面添加一个元素
            while(S.substr(0,s.length())!=s)//反复删除s的第一个元素，直到s成为S的前缀
                s = s.substr(1);
            nex[i][j] = s.length();
        }
    }

    for (int i=0;i<n;i++)
    {
        for (int x=0;x<k;x++)
        {
            for (int j=0;j<4;j++)
            {
                int t = nex[x][j];
                if (t==k)
                    continue; // 排除违反题目条件的那个特例 
                dp[i+1][t] = (dp[i+1][t] + dp[i][x]) % MOD;
            }
        }
    }
    int ans = 0;
    for (int i=0;i<k;i++)
        ans = (ans + dp[n][i]) % MOD;
    cout << ans << endl;
}

int main()
{
    while(cin>>n>>k>>S)
    {
        init();
        solve();
    }
}