POJ1625 Censored!(AC自动机+DP)

题目问长度m不包含一些不文明单词的字符串有多少个。

依然是水水的AC自动机+DP。。做完后发现居然和POJ2778是一道题,回过头来看都水水的。。。

  • dp[i][j]表示长度i(在自动机转移i步)且后缀状态为自动机第j个结点的合法字符串数
  • dp[0][0]=1
  • 转移转移。。。

注意要用高精度,因为答案最多5050

还有就是要用unsigned char,题目的输入居然有拓展的ASCII码,编码128-255。

 #include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
int tn,ch[][],fail[],flag[];
void insert(unsigned char *s){
int x=;
for(int i=; s[i]; ++i){
int y=s[i];
if(ch[x][y]==) ch[x][y]=++tn;
x=ch[x][y];
}
flag[x]=;
}
void init(){
memset(fail,,sizeof(fail));
queue<int> que;
for(int i=; i<; ++i){
if(ch[][i]) que.push(ch[][i]);
}
while(!que.empty()){
int x=que.front(); que.pop();
for(int i=; i<; ++i){
if(ch[x][i]) que.push(ch[x][i]),fail[ch[x][i]]=ch[fail[x]][i],flag[ch[x][i]]|=flag[ch[fail[x]][i]];
else ch[x][i]=ch[fail[x]][i];
}
}
}
#define ULL unsigned long long
#define M 1000000000000000000L
struct BigInt{
ULL bit[];
BigInt(int a=){
bit[]=bit[]=bit[]=bit[]=;
bit[]=a;
}
};
BigInt operator+(const BigInt &b1,const BigInt &b2){
BigInt b;
ULL carry=;
for(int i=; i>=; --i){
ULL tmp=b1.bit[i]+b2.bit[i]+carry;
b.bit[i]=tmp%M;
carry=tmp/M;
}
return b;
};
void output(const BigInt &b){
int i=;
while(i< && b.bit[i]==) ++i;
if(i==){
puts("");
return;
}
printf("%llu",b.bit[i]);
for(int j=i+; j<; ++j){
printf("%018llu",b.bit[j]);
}
putchar('\n');
}
int main(){
BigInt one();
unsigned char str[];
int n,m,p;
while(~scanf("%d%d%d",&n,&m,&p)){
scanf("%s",str);
bool vis[]={};
for(int i=; str[i]; ++i) vis[str[i]]=;
tn=;
memset(ch,,sizeof(ch));
memset(flag,,sizeof(flag));
for(int i=; i<p; ++i){
scanf("%s",str);
insert(str);
}
init();
BigInt **d=new BigInt*[m+];
for(int i=; i<=m; ++i) d[i]=new BigInt[tn+];
d[][]=one;
for(int i=; i<m; ++i){
for(int j=; j<=tn; ++j){
if(flag[j]) continue;
for(int k=; k<; ++k){
if(!vis[k] || flag[ch[j][k]]) continue;
d[i+][ch[j][k]]=d[i+][ch[j][k]]+d[i][j];
}
}
}
BigInt res=;
for(int i=; i<=tn; ++i) res=res+d[m][i];
output(res);
for(int i=; i<=m; ++i) delete[] d[i];
delete[] d;
}
return ;
}
上一篇:nyoj 1022 最少步数【优先队列+广搜】


下一篇:HDU2296 Ring(AC自动机+DP)