「luogu3706」[SDOI2017]硬币游戏

给题解跪辣！

这做法太神了。。

 1 #include<bits/stdc++.h>
 2 #define db double
 3 using namespace std;
 4 const int N=310,M=90010;
 5 const db eps=1e-10;
 6 int n,m;
 7 int ch[M][2],fail[M],v[M],dis[M],tot,pos[N];
 8 db a[N][N],p2[N];
 9 char s[N]; 
10 vector<int>bel[M];
11 void insert(char* str,int id){
12     int now=0;
13     for(int i=0;i<m;i++){
14         bel[now].push_back(id);
15         bool b=str[i]=='H';
16         if(!ch[now][b]) ch[now][b]=++tot,dis[tot]=dis[now]+1;
17         now=ch[now][b];
18     }
19     v[now]=id,pos[id]=now;
20     return;
21 }
22 void build_ac(){
23     queue<int>q;
24     for(int i=0;i<=1;i++) if(ch[0][i]) q.push(ch[0][i]);
25     int x,nxt;
26     while(!q.empty()){
27         x=q.front();q.pop();
28         for(int i=0;i<=1;i++)if(ch[x][i]){
29             q.push(ch[x][i]);
30             nxt=fail[x];
31             while(nxt&&!ch[nxt][i]) nxt=fail[nxt];
32             fail[ch[x][i]]=ch[nxt][i];
33         }
34     }
35     return;
36 }
37 void gauss(){
38     for(int i=1;i<=n;i++){
39         int nxt=i;
40         for(int j=i+1;j<=n;j++)
41             if(fabs(a[j][i])>fabs(a[nxt][i])) nxt=j;
42         if(nxt!=i) for(int j=i;j<=n+1;j++) swap(a[i][j],a[nxt][j]);
43         for(int j=i+1;j<=n;j++){
44             db f=a[j][i]/a[i][i];
45             for(int k=i;k<=n+1;k++) a[j][k]-=a[i][k]*f;
46         }
47     }
48     for(int i=n;i>=1;i--){
49         a[i][n+1]/=a[i][i];
50         for(int j=i-1;j>=1;j--) a[j][n+1]-=a[j][i]*a[i][n+1];
51     }
52     return;
53 }
54 int main(){
55     scanf("%d%d",&n,&m);
56     p2[0]=1;
57     for(int i=1;i<=m;i++) p2[i]=p2[i-1]/2;
58     for(int i=1;i<=n;i++) a[i][n+1]=-p2[m];
59     for(int i=1;i<=n;i++){scanf("%s",s);insert(s,i);}
60     build_ac();
61     for(int i=1;i<=n;i++){
62         a[i][i]=1;
63         int now=fail[pos[i]];
64         while(now){
65             for(int j=0;j<bel[now].size();j++){
66                 int pre=bel[now][j];
67                 a[pre][i]+=p2[m-dis[now]];
68             }
69             now=fail[now];
70         }
71     }
72     for(int i=1;i<=n;i++) a[n+1][i]=1;
73     a[n+1][n+2]=1;
74     n++;
75     gauss();
76     for(int i=1;i<n;i++) printf("%.10lf\n",a[i][n+1]);
77     return 0;
78 }