AC自动机（1）

Description

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀). 

 

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串. 

注意:本题只有一组测试数据,处理到文件结束. 

 

Output

对于每个提问,给出以该字符串为前缀的单词的数量. 

 

Sample Input

banana band bee absolute acm ba b band abc

 

Sample Output

2 3 1 0

 

链接好的代码（结构体）：http://m.blog.csdn.net/blog/sr19930829/26866069（同下）

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <malloc.h>
using namespace std;
char str[12];
const int maxn=26;//孩子节点的最大个数，如果是只有26个字母，就用26就可以了。

struct Trie//树的结构体
{
    int cnt;//保存某个字母出现的次数
    Trie *next[maxn];//每一个节点对应着多少个孩子，如果只有26个字母，就用26就可以了
};

Trie root;

/*void init(Trie t)
{
    for(int i=0;i<26;i++)
        t.next[i]=NULL;
}*///不需要单独对根节点初始化

void CreateTrie(char *str)
{
    int len=strlen(str);
    Trie *p=&root,*q;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)//第一次遇到
        {
            q=(Trie*)malloc(sizeof(Trie));
            q->cnt=1;//此处一开始写错，写成了q->cnt++;
            for(int i=0;i<maxn;i++)
                q->next[i]=NULL;//初始化非空节点的孩子节点
            p->next[id]=q;//在树中填上
            p=p->next[id];//此时的P是不为空的节点
        }
        else
        {
            p->next[id]->cnt++;//不是第一次遇到，个数++
            p=p->next[id];
        }
    }
}

int find(char *str)
{
    int len=strlen(str);
    Trie *p=&root;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        p=p->next[id];//一直向下走。
        if(p==NULL)//找不到该单词，一开始此处写错了，写成了p->next[id]==NULL
            return 0;
    }
    return p->cnt;
}
int main()
{
    while(gets(str)&&str[0]!='\0')
    {
        CreateTrie(str);
    }
    while(scanf("%s",str)!=EOF)
    {
        printf("%d\n",find(str));
    }
    return 0;
}

我的代码（测试样例能过，提交过不了）：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int NODE=1e5+10,CH=26;
int ch[NODE][CH],sz,val[NODE],cal[NODE][CH];

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

int node()
{
    memset(ch[sz],0,sizeof(ch[sz]));
    val[sz]=0;
    return sz++;
}

void init()
{
    sz=0;
    node();
}

void insert(char *s,int v)
{
    int u=0;
    for(;*s;s++)
    {
        int c=idx(*s);
        if(!ch[u][c])
        ch[u][c]=node();
        else
        cal[u][c]++;

        u=ch[u][c];
    }
    val[u]=v;
}

int main()
{
    int ca=1;
    char k[20],*s;
    init();
    memset(cal,0,sizeof(cal));
    while(1)
    {
        gets(k);
        if(strcmp(k,"\0")==0)
        break;
        insert(k,ca++);
    }

    while(scanf("%s",k)!=NULL)
    {
       s=k;
       int c,x=0,u=0,flag=0;
       for(;*s;s++)
       {
           c=idx(*s);
           u=x;
           if(!ch[u][c])
           {
               flag=1;
               break;
           }
           else
           x=ch[u][c];
       }
       if(flag)
       printf("0\n");
       else
       printf("%d\n",cal[u][c]+1);
    }
    return 0;
}