hdu 3065 病毒侵袭持续中（AC automaton)

题目：http://acm.hdu.edu.cn/showproblem.php?pid=3065

每个子串是不同的，求出在源码串中各个子串出现的次数，按照输入子串的顺序将出现次数大于0的输出。
例如：

Sample Input

3AABBCCooxxCC%dAAAoen....END

 

Sample Output

AA: 2CC: 1
Hint
Hit：题目描述中没有被提及的所有情况都应该进行考虑。比如两个病毒特征码可能有相互包含或者有重叠的特征码段。计数策略也可一定程度上从Sample中推测。
 

AC自动机模板中的insert函数代码p->sum++;统计的是子串出现的次数（可能出现相同的子串，那时sum就大于1了）。本题的子串每个只出现一次，需要找到在源码串中每个子串出现的次数，且要输出对应的子串。所以要建立好字符串和结点的映射关系。输出的sum也要灵活处理。事实上，tag的是否标记就决定了源码串中子串是否可以重叠，标记了则不可以重叠，没有标记则能重叠。话说回来，AC自动机应该是我见过比较耗内存的算法，每次提交都和内存限制有个亲密的接触。

#include <iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<queue>using namespace std;int head,tail;const int nn=60; //122-65=57int N;char str[2000010],keyword[55];  char s[1000][55];struct node{    int sum,id;    node *next[nn],*fail;    bool operator <(const node other)const {        return id>other.id;    }    node(){ //inital node        fail=NULL;        id=sum=0;        memset(next,0,sizeof(next)); //here    }}*q[55*1000];node *root;void insert_t(char str[],int dex){    int temp,len;    node *p=root;    len=strlen(str);    for(int i=0;i<len;i++){        temp=str[i]-65;        if(p->next[temp]==NULL)p->next[temp]=new node();        p=p->next[temp];    }    p->id=dex;    strcpy(s[dex],str);    p->sum++;}void build_ac(){  //inital fail point    q[tail++]=root;    while(head<tail){        node *p=q[head++],*temp=NULL;        for(int i=0;i<nn;i++){            if(p->next[i]!=NULL){                if(p==root)p->next[i]->fail=root;                else {                temp=p->fail;                while(temp){                    if(temp->next[i]){                        p->next[i]->fail=temp->next[i];                        break;                    }                    temp=temp->fail;                }                if(!temp)p->next[i]->fail=root;                }                q[tail++]=p->next[i];            }        }    }}priority_queue<node> que;void query(){      int i,dex,len=strlen(str);    node *p=root;    for(i=0;i<len;i++){        dex=str[i]-65;        if(dex<0||dex>57){            p=root;            continue;        }        while(p->next[dex]==NULL&&p!=root)p=p->fail;        p=p->next[dex];        if(!p)p=root;        node *temp=p;        while(temp!=root&&temp->sum!=0){            que.push(*temp);            temp=temp->fail;        }    }}int main(){    int t;    while(cin>>N){        head=tail=0;        root=new node();        int i,j,total=0;        for(i=0;i<N;i++){            scanf("%s",keyword);            insert_t(keyword,i+1);        }        build_ac();        scanf("%s",str);        query();        while(!que.empty()){            node temp1=que.top();            que.pop();            if(!que.empty()){                node temp2=que.top();                //que.pop();                while(temp1.id==temp2.id){                    temp2.sum+=temp1.sum;  //处理sum                    temp1=temp2;                    que.pop();                    if(!que.empty()){                        temp2=que.top();                    }                    else break;                }            }            printf("%s: %d\n",s[temp1.id],temp1.sum);        }    }    return 0;}