codeforce 271D Good Substrings (后缀自动机+dp)

题意：

给出一个串，和一个01串，01串只有26个字符，每个字符对应26个英文字母表示这个字符是否是好的，1表示好的，0表示不好的。所给的串中多少个不同子串至多包含k个坏字母。

题解：

这题没想到用dp来做，一直去想性质，其实用dp做挺简单的。dp[i][j]表示在后缀动机上的i点，现在有j个坏字母对应子串的个数。明显:

dp[i][k]+=dp[next[i][j]][k] if(alp[j]==1)

dp[i][k]+=dp[next[i][j]][k-1] if(alp[j]==0)

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define B(x) (1<<(x))
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const int MOD=2012;
const int maxn=1600;
const int SIZE=maxn<<1;
const int type=26;
int Next[SIZE][type],fa[SIZE],len[SIZE];
int cnt[SIZE],pos[SIZE];
int tol,last;
char str[maxn];
int alp[30],dp[SIZE][maxn];

int newNode(int x){

    len[tol]=x;
    fa[tol]=-1;
    for(int i=0;i<type;i++){
        Next[tol][i]=-1;
    }
    return tol++;
}

void Init(){
    tol=0;
    last=newNode(0);
}

void add(int k){

    int now=last;
    int end=newNode(len[now]+1);
    while(now!=-1&&Next[now][k]==-1){
        Next[now][k]=end;
        now=fa[now];
    }
    if(now==-1) fa[end]=0;
    else{
        int nxt=Next[now][k];
        if(len[nxt]==len[now]+1) fa[end]=nxt;
        else{
            int cnxt=newNode(len[now]+1);
            for(int i=0;i<type;i++) Next[cnxt][i]=Next[nxt][i];
            fa[cnxt]=fa[nxt];
            fa[nxt]=fa[end]=cnxt;
            while(now!=-1&&Next[now][k]==nxt){
                Next[now][k]=cnxt;
                now=fa[now];
            }
        }
    }
    last=end;
}

void Insert(char T[]){

    for(int i=0;T[i];i++)
        add(T[i]-'a');
}

void DP(int m){
    for(int i=0;i<tol;i++)cnt[i]=0;
    for(int i=0;i<tol;i++)cnt[len[i]]++;
    for(int i=1;i<tol;i++)cnt[i]+=cnt[i-1];
    for(int i=tol-1;i>=0;i--)pos[--cnt[len[i]]]=i;
    memset(dp,0,sizeof dp);
    for(int i=1;i<tol;i++)dp[i][0]=1;///0不要，因为没有空串
    for(int i=tol-1;i>=0;i--){
        int p=pos[i];
        for(int j=0;j<type;j++){
            for(int k=0;k<=m;k++){
                if(Next[p][j]==-1)continue;
                if(alp[j]==1)
                    dp[p][k]+=dp[Next[p][j]][k];
                else if(k>0)
                    dp[p][k]+=dp[Next[p][j]][k-1];
            }
        }
    }
    int ans=0;
    for(int i=0;i<=m;i++)
        ans+=dp[0][i];
    printf("%d\n",ans);
}

int main(){
/*
#define ON 1
#ifdef ON
    freopen("E:\\read.txt","r",stdin);
#endif // ON
//*/
    int m;
    while(scanf("%s",str)!=EOF){
        Init();
        Insert(str);
        scanf("%s",str);
        for(int i=0;str[i];i++)alp[i]=str[i]-'0';
        scanf("%d",&m);
        DP(m);
    }
    return 0;
}