题目描述:给n个可用字母,组成长为m的串s,其中有w个子串t是不能在这些串s中出现的,请问这样的串s有多少种?
思路:先建立一个ac自动机,并且标记每个w的结束节点end=1,同时把fail指向end=1的点的end也改写为1。现在要计算从root走m步(不能经过end=1的点)一共有多少种走法。采用DP的写法,记dp[step][i]表示从root走step步可以到j点。如果i能一步走到j,那么dp[step][j]=dp[step-1][i](所有满足的i都加进来)。
//POJ 1625 关键代码段
void build(){
queue <int> Q;
fail[root] = root;
for(int i = 0;i < n;i++)
if(nxt[root][i] == -1)
nxt[root][i] = root;
else{
fail[nxt[root][i]] = root;
Q.push(nxt[root][i]);
}
while(!Q.empty()){
int now = Q.front(); Q.pop();
for(int i = 0;i < n;i++){
if(nxt[now][i] == -1)
nxt[now][i] = nxt[fail[now]][i];
else{
fail[nxt[now][i]] = nxt[fail[now]][i];
end[nxt[now][i]] |= end[nxt[fail[now]][i]];//end的重标
Q.push(nxt[now][i]);
}
}
}
}
void solve(int step)
{
memset(dp, 0, sizeof(dp));
dp[0][0][0] = 1; //大数一定会爆,用数组模拟
for(int k = 1;k <= step;k++){ //注意了,一定是把步数放在最外面统计
for(int i = 0;i < L;i++){
if(end[i]) continue; //end是不能实现字段的结束标识,不可到达
for(int j = 0;j < n;j++){
int thi = nxt[i][j];
if(end[thi]) continue;
ADD(dp[k][thi], dp[k-1][i]);
}
}
}
for(int i = 1;i < L;i++){ //这里是把所有结点上的可能加起来,不要重复加结点0
if(end[i]) continue;
ADD(dp[step][0], dp[step][i]);
}
pin(dp[step][0]);
}