zoj 3228 Searching the String

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=3441

trie       这是一个自动机的题 不过trie就可以解决

先对长串进行处理 由于后面出现的字符串最长为6 所以以长串每个位置的字符为起点用长度1-6的子串进行建树

建树时 对树里面的节点要维护 到此节点位置的字符串在原串中出现的位置（多个） 并更新出现的个数和不允许重叠的个数

然后输入短串时就好处理了

代码：

#include <iostream>

#include <algorithm>

#include <cmath>

#include <cstdio>

#include <cstring>

#include <string>

#include <vector>

#include <queue>

#include <stack>



using namespace std;



const int N=1005;

const int M=100010;

const int K=26;

int I;

struct node

{

    int v,next;

}edge[M*6];

struct nodeTrie

{

    int next[K];

    int head;

    int len;

    void initialize()

    {

        head=-1;

        len=0;

        memset(next,-1,sizeof(next));

    }

}trie[M*6];

int cnt,root;

char s[10];

char ls[M];

int num[M*6];

int getNewNode()

{

    ++cnt;

    trie[cnt].initialize();

    return cnt;

}

void addWord(int p,int l,int r)

{

    for(int i=l;i<r;++i)

    {

        if(trie[p].next[ls[i]-'a']==-1)

        trie[p].next[ls[i]-'a']=getNewNode();

        p=trie[p].next[ls[i]-'a'];

        num[I]=1;

        ++trie[p].len;

        edge[I].v=l;

        edge[I].next=trie[p].head;

        trie[p].head=I;

        for(int t=trie[p].head;t!=-1;t=edge[t].next)

        if(edge[t].v+(i-l+1)<=l)

        {num[I]=num[t]+1;break;}

        ++I;

    }

}

void init(int n)

{

    cnt=-1;I=0;

    root=getNewNode();

    for(int i=0;i<n;++i)

    addWord(root,i,min(i+6,n));

}

int searchWord(int p,char *s)

{

    for(int i=0;s[i]!='\0';++i)

    {

        if(trie[p].next[s[i]-'a']==-1)

        return 0;

        p=trie[p].next[s[i]-'a'];

    }

    return p;

}

int findNum(char *s,int flag)

{

    int p=searchWord(root,s);

    if(p==0) return 0;

    if(flag==0)

    return trie[p].len;

    if(trie[p].len<=1)

    return trie[p].len;

    return num[trie[p].head];

}

int main()

{

    //freopen("data.in","r",stdin);

    int ca=0;

    while(gets(ls))

    {

        ++ca;

        init(strlen(ls));

        int n;

        scanf("%d",&n);

        printf("Case %d\n",ca);

        while(n--)

        {

            int flag;

            scanf("%d ",&flag);

            gets(s);

            printf("%d\n",findNum(s,flag));

        }

        printf("\n");

        scanf(" ");

    }

    return 0;

}