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;
}