ZOJ 2619 && HDU 3058 (一)

两道AC自动机+高消的求期望问题

先说HDU 3058

[cpp]
#include
#include
#include
#include
#include
#define N 110
using namespace std;
int next[N][8],flag[N],fail[N],pos,tot,id[N];
int n;
char s[11];
double mat[110][110];

int newnode(){
memset(next[pos],0,sizeof(next[pos]));
flag[pos]=fail[pos]=0;
return pos++;
}

void insert(){
int p=0,i;
for(i=0;s[i];i++){
int k=s[i]-'A';
p=next[p][k] next[p][k]:next[p][k]=newnode();
}
flag[p]=1;
}

void makefail(){
queueq;
q.push(0);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=0;i int v=next[u][i];
if(v==0) next[u][i]=next[fail[u]][i];
else q.push(v);
if(u && v){
fail[v]=next[fail[u]][i];
flag[v]|=flag[fail[v]];
}
}
}
}

void make_mat(){
int i,j;
tot=0;
memset(id,0,sizeof(id));
for(i=0;i if(flag[i]==0) id[i]=tot++;
memset(mat,0,sizeof(mat));
for(i=0;i if(flag[i]==1)continue;
mat[id[i]][tot]=-n;
mat[id[i]][id[i]]=-n;
for(j=0;j int k=next[i][j];
if(flag[k]==1)continue;
mat[id[i]][id[k]]+=1.0;
}
}
}

bool gauss(){
int i,j,row,id;
double maxx;
for(row=0;row maxx=fabs(mat[row][row]);id=row;
for(i=row+1;i if(fabs(mat[i][row]>maxx)){
maxx=fabs(mat[i][row]);
id=i;
}
if(maxx<1e-11) return 0; //我一开始这里精度设置1e-8就WA死了。。。。。。
if(id!=row){
for(j=0;j<=tot;j++)
swap(mat[row][j],mat[id][j]);
}
for(i=row+1;i for(j=tot;j>=row;j--)
mat[i][j]-=mat[i][row]/mat[row][row]*mat[row][j];
}
}
for(i=tot-1;i>=0;i--){
for(j=i+1;j mat[i][tot]-=mat[i][j]*mat[j][j];
mat[i][i]=mat[i][tot]/mat[i][i];
}
return 1;
}

int main(){
int t,T,m,i;
scanf("%d",&T);
for(t=1;t<=T;t++){
scanf("%d %d",&n,&m);
pos=0,newnode();
for(i=1;i<=m;i++){
scanf("%s",s);
insert();
}
makefail();
make_mat();
gauss();
printf("%.2lf\n",mat[0][0]);
}
}

#include
#include
#include
#include
#include
#define N 110
using namespace std;
int next[N][8],flag[N],fail[N],pos,tot,id[N];
int n;
char s[11];
double mat[110][110];

int newnode(){
memset(next[pos],0,sizeof(next[pos]));
flag[pos]=fail[pos]=0;
return pos++;
}

void insert(){
int p=0,i;
for(i=0;s[i];i++){
int k=s[i]-'A';
p=next[p][k] next[p][k]:next[p][k]=newnode();
}
flag[p]=1;
}

void makefail(){
queueq;
q.push(0);
while(!q.empty()){
int u=q.front();
q.pop();
for(int i=0;i int v=next[u][i];
if(v==0) next[u][i]=next[fail[u]][i];
else q.push(v);
if(u && v){
fail[v]=next[fail[u]][i];
flag[v]|=flag[fail[v]];
}
}
}
}

void make_mat(){
int i,j;
tot=0;
memset(id,0,sizeof(id));
for(i=0;i if(flag[i]==0) id[i]=tot++;
memset(mat,0,sizeof(mat));
for(i=0;i if(flag[i]==1)continue;
mat[id[i]][tot]=-n;