ZOJ 3228 Searching the String(自动机)

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=3441

题意：给定一个串s以及N个子串，问每个子串在s中出现的次数。有些子串允许在s中有重叠有些不允许。

思路：自动机：每个节点记录到达次数，type为1时记录前一次到达时是啥时候。

const int N=200005;

struct node
{
    int next[26],fail,cnt[2],pre,len;

    void init()
    {
        clr(next,-1);
        fail=-1;
        cnt[0]=cnt[1]=0;
        pre=-1;
        len=0;
    }
};

node a[N];
int e,n,*mp[N];


void insert(char s[],int t,int id)
{
    int i,k,p=0;
    for(i=0;s[i];i++)
    {
        k=s[i]-'a';
        if(a[p].next[k]==-1)
        {
            a[e].init();
            a[p].next[k]=e++;
        }
        p=a[p].next[k];
    }
    a[p].len=i;
    mp[id]=&(a[p].cnt[t]);
}


void build()
{
    queue<int> Q;
    int i,j,k,t,p,q;
    Q.push(0);
    while(!Q.empty())
    {
        k=Q.front();
        Q.pop();

        for(i=0;i<26;i++)
        {
            if(a[k].next[i]!=-1)
            {
                p=a[k].next[i];
                q=a[k].fail;
                while(q!=-1&&a[q].next[i]==-1) q=a[q].fail;
                if(q==-1) a[p].fail=0;
                else a[p].fail=a[q].next[i];
                Q.push(p);
            }
            else
            {
                q=a[k].fail;
                while(q!=-1&&a[q].next[i]==-1) q=a[q].fail;
                if(q==-1) a[k].next[i]=0;
                else a[k].next[i]=a[q].next[i];
            }
        }
    }
}

char s[N];


void find(char s[])
{
    int p=0,i,j,k,t;
    for(i=0;s[i];i++)
    {
        k=s[i]-'a';
        p=a[p].next[k];
        t=p;
        while(t!=-1)
        {
            if(a[t].len>0)
            {
                a[t].cnt[0]++;
                if(a[t].pre==-1||i-a[t].pre>=a[t].len)
                {
                    a[t].cnt[1]++;
                    a[t].pre=i;
                }
            }
            t=a[t].fail;
        }
    }
}

int main()
{
    int num=0;
    while(scanf("%s",s)!=-1)
    {
        a[0].init();
        e=1;
        RD(n);
        char str[15];
        int i,j;
        FOR0(i,n) RD(j),RD(str),insert(str,j,i);
        build();
        find(s);
        printf("Case %d\n",++num);
        FOR0(i,n) PR(*mp[i]);
        puts("");
    }
}