hdu 2222——Keywords Search

题意：给出几个关键字，在给出一个主串，要求主串里面包含多少个关键字

思路：AC自动机

错误：题意理解不清楚

开始的时候以为可以重复计算，后来改过来还是错。

后来发现是关键字可以重复，这个时候以为每找到一个关键字就算一次。后来才发现，如果关键字是2个a，主串里面有一个a 的时候，应当直接在答案上加上两个a，而不是减去一个a再等着找下一个a

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;

const int maxnode=10000*50+10;
const int sigma_size=26;

int f[maxnode];
int last[maxnode];

int idx(char c){return c-'a';}

struct Trie
{
        int ch[maxnode][sigma_size];
        int val[maxnode];
        int cnt[maxnode];
        int sz;
        int ans;
        Trie(){sz=1;memset(ch[0 ],0,sizeof(ch[0]));ans=0;}

        void init()
        {
                sz=1;memset(ch[0 ],0,sizeof(ch[0]));
                ans=0;
                memset(val,0,sizeof(val));
                memset(cnt,0,sizeof(cnt));
                memset(last,0,sizeof(last));
                memset(f,0,sizeof(f));
        }

        void insert(char *s,int v)
        {
                int u=0;
                int n=strlen(s);
                for(int i=0;i<n;++i)
                {
                        int c=idx(s[i]);
                        if(!ch[u][c]){
                                memset(ch[sz],0,sizeof(ch[sz]));
                                val[sz]=0;
                                ch[u][c]=sz++;
                        }
                        u=ch[u][c];
                }
                val[u]=v;
                cnt[u]++;
        }

        void getfail(){
                queue<int> q;
                f[0]=0;
                for(int c=0;c<sigma_size;++c)
                {
                        int &u=ch[0][c];
                        if(u){f[u]=0;q.push(u);last[u]=0;}
                }
                while(!q.empty())
                {
                        int r=q.front();q.pop();
                        for(int c=0;c<sigma_size;++c)
                        {
                                int u=ch[r][c];
                                if(!u)continue;
                                q.push(u);
                                int v=f[r];
                                while(v && !ch[v][c])v=f[v];
                                f[u]=ch[v][c];
                                last[u]= val[f[u]]?f[u]:last[f[u]];
                        }
                }
        }

        void find(char* T)
        {
                int n=strlen(T);
                int j=0;
                for(int i=0;i<n;++i)
                {
                        int c=idx(T[i]);
                        while(j&&!ch[j][c])j=f[j];
                        j=ch[j][c];
                        if(val[j])print(j);
                        else if(last[j])print(last[j]);
                }
        }

        void print(int j)
        {
                if(j){
                        if(cnt[j]){ans+=cnt[j];cnt[j]=0;}
                        print(last[j]);
                }
        }

};

Trie tr;

int main()
{
//freopen("data.txt","r",stdin);
        int T;
        scanf("%d",&T);
        while(T--)
        {
                tr.init();
                int n;
                scanf("%d",&n);
                for(int i=0;i<n;++i)
                {
                        char str[55];
                        scanf("%s",str);
                        tr.insert(str,i+1);
                }
                tr.getfail();
                char s[1000005];
                scanf("%s",s);
                tr.find(s);
                printf("%d\n",tr.ans);
        }
        return 0;
}

--------------------------------------------------------------------------------------------------------------------------------------------------------------------------

题目的数据好像加强了。。原来AC过的，今天再提交就MLEL了。

修改了两个地方

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;

const int maxnode=10000*50+10;
const int sigma_size=26;

int f[maxnode];
int last[maxnode];

int idx(char c){return c-'a';}

struct Trie
{
        int ch[maxnode][sigma_size];
        int val[maxnode];
        int cnt[maxnode];
        int sz;
        int ans;
        Trie(){sz=1;memset(ch[0 ],0,sizeof(ch[0]));ans=0;}

        void init()
        {
                sz=1;memset(ch[0 ],0,sizeof(ch[0]));
                ans=0;
                memset(val,0,sizeof(val));
//                memset(cnt,0,sizeof(cnt));
//                memset(last,0,sizeof(last));
//                memset(f,0,sizeof(f));这里的三行全部给注释掉了
        }

        void insert(char *s,int v)
        {
                int u=0;
                int n=strlen(s);
                for(int i=0;i<n;++i)
                {
                        int c=idx(s[i]);
                        if(!ch[u][c]){
                                memset(ch[sz],0,sizeof(ch[sz]));
                                val[sz]=0;
                                ch[u][c]=sz++;
                        }
                        u=ch[u][c];
                }
                val[u]=v;
                cnt[u]++;
        }

        void getfail(){
                queue<int> q;
                f[0]=0;
                for(int c=0;c<sigma_size;++c)
                {
                        int &u=ch[0][c];
                        if(u){f[u]=0;q.push(u);last[u]=0;}
                }
                while(!q.empty())
                {
                        int r=q.front();q.pop();
                        for(int c=0;c<sigma_size;++c)
                        {
                                int u=ch[r][c];
                                if(!u)continue;
                                q.push(u);
                                int v=f[r];
                                while(v && !ch[v][c])v=f[v];
                                f[u]=ch[v][c];
                                last[u]= val[f[u]]?f[u]:last[f[u]];
                        }
                }
        }

        void find(char* T)
        {
                int n=strlen(T);
                int j=0;
                for(int i=0;i<n;++i)
                {
                        int c=idx(T[i]);
                        while(j&&!ch[j][c])j=f[j];
                        j=ch[j][c];
                        if(val[j])print(j);
                        else if(last[j])print(last[j]);
                }
        }

        void print(int j)
        {
                if(j){
                        if(cnt[j]){ans+=cnt[j];cnt[j]=0;
                        val[j]=0;//这里加上了这个
                        }
                        print(last[j]);
                }
        }

};

Trie tr;

int main()
{
//freopen("data.txt","r",stdin);
        int T;
        scanf("%d",&T);
        while(T--)
        {
                tr.init();
                int n;
                scanf("%d",&n);
                for(int i=0;i<n;++i)
                {
                        char str[55];
                        scanf("%s",str);
                        tr.insert(str,i+1);
                }
                tr.getfail();
                char s[1000005];
                scanf("%s",s);
                tr.find(s);
                printf("%d\n",tr.ans);
        }
        return 0;
}