【HDU3341】 Lost's revenge （AC自动机+状压DP）

  1 #include<cstdio>
  2 #include<cstdlib>
  3 #include<cstring>
  4 #include<iostream>
  5 #include<algorithm>
  6 #include<queue>
  7 using namespace std;
  8 #define Maxn 510
  9 #define Maxl 15000
 10 #define INF 0xfffffff
 11 #define Mod 20090717
 12 
 13 int n;
 14 char s[210];
 15 int v[5],k[5],ln;
 16 
 17 struct node
 18 {
 19     // int cnt;
 20     int fail,mark;
 21     int son[5];
 22 }t[Maxn];int tot;
 23 int num;
 24 int p[210];
 25 
 26 void upd(int x)
 27 {
 28     // t[x].cnt=0;
 29     t[x].mark=0;
 30     memset(t[x].son,0,sizeof(t[x].son));
 31 }
 32 
 33 int mymin(int x,int y) {return x<y?x:y;}
 34 int mymax(int x,int y) {return x>y?x:y;}
 35 
 36 void read_trie()
 37 {
 38     scanf("%s",s+1);
 39     int len=strlen(s+1);
 40     int now=0;
 41     for(int i=1;i<=len;i++)
 42     {
 43         if(s[i]==‘C‘) s[i]=‘B‘;
 44         else if(s[i]==‘G‘) s[i]=‘C‘;
 45         else if(s[i]==‘T‘) s[i]=‘D‘;
 46         int ind=s[i]-‘A‘+1;
 47         if(!t[now].son[ind]) 
 48         {
 49             t[now].son[ind]=++tot;
 50             upd(tot);
 51         }
 52         now=t[now].son[ind];
 53         if(i==len) t[now].mark++;
 54     }
 55 }
 56 
 57 queue<int > q;
 58 void build_AC()
 59 {
 60     while(!q.empty()) q.pop();
 61     q.push(0);
 62     while(!q.empty())
 63     {
 64         int x=q.front();q.pop();
 65         for(int i=1;i<=4;i++) 
 66         {
 67             if(t[x].son[i])
 68             {
 69                 t[t[x].son[i]].fail=x?t[t[x].fail].son[i]:0;
 70                 q.push(t[x].son[i]);
 71             }
 72             else t[x].son[i]=t[t[x].fail].son[i];
 73         }
 74         t[x].mark+=t[t[x].fail].mark;
 75     }
 76 }
 77 
 78 int f[Maxn][Maxl];
 79 int a[5];
 80 void dp()
 81 {
 82     memset(f,-1,sizeof(f));
 83     f[0][0]=0;
 84     int ans=0;
 85      for(int l=0;l<=ln;l++) //选取的字母个数
 86     for(int i=0;i<=tot;i++) //走到节点i
 87       for(a[1]=mymax(l-v[4]-v[3]-v[2],0);a[1]<=mymin(v[1],l);a[1]++)//if(f[i-1][j][l]!=0)
 88        for(a[2]=mymax(l-v[4]-v[3]-a[1],0);a[2]<=mymin(v[2],l);a[2]++)
 89         for(a[3]=mymax(l-v[4]-a[1]-a[2],0);a[3]<=mymin(v[3],l);a[3]++)
 90         {
 91             a[4]=l-a[1]-a[2]-a[3];
 92             int now=a[1]*k[1]+a[2]*k[2]+a[3]*k[3]+a[4];
 93             if(f[i][now]==-1) continue; 
 94             for(int j=1;j<=4;j++) if(a[j]+1<=v[j])
 95             {
 96                 // if(t[t[i].son[j]].mark) 
 97                  f[t[i].son[j]][now+k[j]]=mymax(f[t[i].son[j]][now+k[j]],f[i][now]+t[t[i].son[j]].mark);
 98                 // else f[t[i].son[j]][now+k[j]]=mymax(f[t[i].son[j]][now+k[j]],f[i][now]);
 99                 ans=mymax(ans,f[t[i].son[j]][now+k[j]]);
100             }
101         }
102     printf("%d\n",ans);
103 }
104 
105 char ss[50];
106 void init()
107 {
108     tot=0;
109     upd(0);
110     for(int i=1;i<=n;i++)
111     {
112         read_trie();
113     }
114     build_AC();
115     scanf("%s",ss+1);
116     ln=strlen(ss+1);
117     v[1]=v[2]=v[3]=v[4]=0;
118     for(int i=1;i<=ln;i++)
119     {
120         if(ss[i]==‘C‘) ss[i]=‘B‘;
121         else if(ss[i]==‘G‘) ss[i]=‘C‘;
122         else if(ss[i]==‘T‘) ss[i]=‘D‘;
123         v[ss[i]-‘A‘+1]++;
124     }
125     int maxx=(v[1]+1)*(v[2]+1)*(v[3]+1)*(v[4]+1);
126     k[1]=(v[2]+1)*(v[3]+1)*(v[4]+1);
127     k[2]=(v[3]+1)*(v[4]+1);
128     k[3]=(v[4]+1);k[4]=1;
129 }
130 
131 int main()
132 {
133     int kase=0;
134     while(1)
135     {
136         scanf("%d",&n);
137         if(n==0) break;
138         init();
139         printf("Case %d: ",++kase);
140         dp();
141     }
142     return 0;
143 }