HDU P3341 Lost's revenge 题解+数据生成器

Lost and AekdyCoin are friends. They always play "number game"(A boring game based on number theory) together. We all know that AekdyCoin is the man called "nuclear weapon of FZU,descendant of Jingrun", because of his talent in the field of number theory. So Lost had never won the game. He was so ashamed and angry, but he didn't know how to improve his level of number theory.

One
noon, when Lost was lying on the bed, the Spring Brother poster on the wall(Lost
is a believer of Spring Brother) said hello to him! Spring Brother said, "I'm
Spring Brother, and I saw AekdyCoin shames you again and again. I can't bear my
believers were being bullied. Now, I give you a chance to rearrange your gene
sequences to defeat AekdyCoin!".

It's soooo crazy and unbelievable to
rearrange the gene sequences, but Lost has no choice. He knows some genes called
"number theory gene" will affect one "level of number theory". And two of the
same kind of gene in different position in the gene sequences will affect two
"level of number theory", even though they overlap each other. There is nothing
but revenge in his mind. So he needs you help to calculate the most "level of
number theory" after rearrangement.

　　　　　　　　　　　　　　——by HDU;

http://acm.hdu.edu.cn/showproblem.php?pid=3341

---

---

就是，有N个由'A'C'G'T'构成的KEY串，给一个可以改变顺序的word串，求最多匹配（KEY可重复使用）

AC自动机+DP

f[i][num0][num1][num2][num3]

表示当前在AC自动机i点，'A'用了num0个，'C'用了num1个......

于是她可以转移到她的所有子节点及子节点的fail链上

但是显然因为num0~3他们的范围的问题（你只能每个都开到word串的长这么大，就是40），这样会MLE（自己算算咯）

发现数组开的很冗余；

于是考虑状态压缩（还是hash？？？）

反正把num0~num3压到数组的一维里，对num0而言每个1在压缩后为该维贡献1（sum0=1），对num1而言每个1在压缩后则为该维贡献1*(num0+1)（sum1=num0+1）,对num1而言每个1在压缩后则为该维贡献1*(num1+1)*sum1（sum2=(num1+1)*sum1）.......

这样把原状态的后4维按权（sum）压到一维里

可以看出，这样压缩后不会有多余的空间；

那么开出的数组就是点数i*(num0+1)*(num1+1)*(num2+1)*(num3+1)（组合数学吧）

最大也就num0~3均取10——也不是很大呢

代码如下：

 #include<cstdio>
 #include<cstring>
 using namespace std;
 struct ss{
     int ch[];
 }x[];
 int n,tot;
 int is_end[],fail[];
 int que[];
 char key[],word[];
 int num[],sum[],f[][];
 int boo(int ,int ,int ,int ,int );
 void bfs_fail();
 int dp();
 int pd(char );
 int main(){
     int i,j,k,len,T=;
     while(){
         scanf("%d",&n);
         if(!n)
             return ;
         tot=;
         memset(is_end,,sizeof(is_end));
         memset(fail,,sizeof(fail));
         memset(x,,sizeof(x));
         for(i=;i<=;i++)
             num[i]=,sum[i]=;
         for(i=;i<=n;i++){
             scanf("%s",key);
             len=strlen(key)-;k=;
             for(j=;j<=len;j++){
                 if(!x[k].ch[pd(key[j])])
                     x[k].ch[pd(key[j])]=++tot;
                 k=x[k].ch[pd(key[j])];
             }
             is_end[k]++;
         }
         bfs_fail();
         scanf("%s",word);len=strlen(word)-;
         for(i=;i<=len;i++)
             num[pd(word[i])]++;
         sum[]=;
         for(i=;i<=;i++)sum[i]=sum[i-]*(num[i-]+);
         printf("Case %d: %d\n",++T,dp());
     }
     return ;
 }
 int boo(int i,int j,int k,int l,int p){
     int x;
     switch (p){
         case :x=i;break;
         case :x=j;break;
         case :x=k;break;
         case :x=l;break;
     }
     if(num[p]-x->=)
         return ;
     return ;
 }
 void bfs_fail(){
     int h=,t=,i,j,k;
     while(h<t){
         ++h;
         for(i=;i<=;i++)
             if(x[que[h]].ch[i]){
                 j=que[h];
                 while(){
                     if(x[j].ch[i]&&j!=que[h]){
                         fail[x[que[h]].ch[i]]=x[j].ch[i];
                         break;
                     }
                     else{
                         if(!j)
                             break;
                         j=fail[j];
                     }
                 }
                 que[++t]=x[que[h]].ch[i];
             }
     }
 }
 int dp(){
     memset(f,-,sizeof(f));f[][]=;
     int i,j,k,l,o,p,q,r,ans=-;
     for(i=;i<=num[];i++)
         for(j=;j<=num[];j++)
             for(k=;k<=num[];k++)
                 for(l=;l<=num[];l++)
                     for(o=;o<=tot;o++)
                         if(f[o][i*sum[]+j*sum[]+k*sum[]+l*sum[]]!=-)
                             for(p=;p<=;p++)
                                 if(x[o].ch[p]&&boo(i,j,k,l,p)){
                                     r=x[o].ch[p];q=;
                                     while(r){
                                         q+=is_end[r];
                                         r=fail[r];
                                     }
                                     r=x[o].ch[p];
                                     while(){
                                         f[r][i*sum[]+j*sum[]+k*sum[]+l*sum[]+sum[p]]=f[r][i*sum[]+j*sum[]+k*sum[]+l*sum[]+sum[p]]>f[o][i*sum[]+j*sum[]+k*sum[]+l*sum[]]+q?f[r][i*sum[]+j*sum[]+k*sum[]+l*sum[]+sum[p]]:f[o][i*sum[]+j*sum[]+k*sum[]+l*sum[]]+q;
                                         if(!r)break;
                                         r=fail[r];
                                     }
                                 }
     q=sum[]*num[]+sum[]*num[]+sum[]*num[]+sum[]*num[];
     for(i=;i<=tot;i++)
         for(j=;j<=q;j++)
             if(ans<f[i][j])
                 ans=f[i][j];
     return ans;
 }
 int pd(char a){
     int i;
     switch (a) {
         case 'A':i=;break;
         case 'C':i=;break;
         case 'G':i=;break;
         case 'T':i=;break;
     }
     return i;
 }

由于本人也在这个题上卡了好久，故放上数据生成器：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<ctime>
#include<cstdlib>
using namespace std;
char s[]={'A','C','G','T'};
int main()
{
    freopen("input.txt","w",stdout);
    srand(time());
    int T=,n,l;
    for(int i=;i<=T;i++){
        n=rand()%+;
        printf("%d\n",n);
        for(int j=;j<=n;j++){
            l=rand()%+;
            for(int k=;k<=l;k++)
                printf("%c",s[rand()%]);
            printf("\0");printf("\n");
        }
        l=rand()%+;
            for(int k=;k<=l;k++)
                printf("%c",s[rand()%]);
            printf("\0");printf("\n");
    }
    printf("");
}

data

祝AC