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
#include
#include
#include
#include
#include
#include
#include
#include
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=0;i--)pos[--cnt[len[i]]]=i;
    memset(dp,0,sizeof dp);
    for(int i=1;i=0;i--){
        int p=pos[i];
        for(int j=0;j0)
                    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;
}





你可能感兴趣的:(字符串—后缀自动机,动态规划—普通dp,字符串)