AC自动机——HDU-3065

AC代码:
#include 
#include 
#include 
#include 
using namespace std;
char model[1005][55];
char str[2000005],tstr[2000005];
int vis[1005];

struct node
{
    node* fail;
    node* nnext[26];
    int ncount;
    node()
    {
        fail=NULL;
        memset(nnext,NULL,sizeof(nnext));
        ncount=-1;
    }
}* root;

void  tire(char* model0,int x)
{
    int index,i=0;
    node* p=root;
    while(model0[i])
    {
        index=model0[i++]-'A';
        if(p->nnext[index]==NULL)
        {
            p->nnext[index]=new node;
        }
        p=p->nnext[index];
    }
    p->ncount=x;
}

void set_fail()
{
    queue q;
    root->fail=NULL;
    q.push(root);
    node* tmp,*p;
    while(!q.empty())
    {
        tmp=q.front();
        q.pop();
        for(int i=0;i<26;i++)
        {
            if(tmp->nnext[i]!=NULL)
            {
                if(tmp==root)
                {
                    tmp->nnext[i]->fail=root;
                }
                else
                {
                    p=tmp->fail;
                    while(p!=NULL)
                    {
                        if(p->nnext[i]!=NULL)
                        {
                            tmp->nnext[i]->fail=p->nnext[i];
                            break;
                        }
                        p=p->fail;
                    }
                    if(p==NULL)
                    {
                        tmp->nnext[i]->fail=root;
                    }
                }
                q.push(tmp->nnext[i]);
            }
        }
    }
}

int find_model(char* str0)
{
    node* p=root;
    node* tmp;
    int cnt=0;
    int i=0;
    int index;
    while(str0[i])
    {
        index=str0[i++]-'A';
        while(p!=root&&p->nnext[index]==NULL)
        {
            p=p->fail;
        }
        p=p->nnext[index];
        if(p==NULL)
            p=root;
        tmp=p;
        while(tmp!=root)
        {
            if(tmp->ncount!=-1)
            {
                vis[tmp->ncount]++;
                cnt++;
            }
            tmp=tmp->fail;
        }
    }
    return cnt;
}


int main()
{
    int n1;
    while(~scanf("%d",&n1))
    {
        memset(vis,0,sizeof(vis));
        root=new node;
        for(int i=0;i='A')
            {
                tstr[flag++]=str[i];
            }
            else
            {
                tstr[flag]='\0';
                cnt+=find_model(tstr);
                flag=0;
            }
        }
        if(flag!=0)
        {
            tstr[flag]='\0';
            cnt+=find_model(tstr);
        }
        if(cnt!=0)
        {
            for(int i=0;i
所遇问题及解决办法:

1:首先,此问题是一道多例问题,这点很不容易注意的到。需要在输入时多出读取并求解。
2:空间超出,由于输入串只有大写字母26个,因此,如果每个AC自动机的节点开128个子节点就太大了。所以,把每次输入中的非大写字母忽略,仅读入大写字母部分的串,在进行查找,累计结果即可。

你可能感兴趣的:(AC自动机——HDU-3065)