题意:
给出一个串,和一个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;
}