题目:
考虑只由’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();
}
}