poj 2945 trie树统计字符串出现次数

用记录附加信息的val数组记录次数即可。

 

trie的原理：每个可能出现的字目给一个编号c，那么整个树就是一个c叉树

ch[u][c]表示 节点u走c边过去之后的节点

 

PS：trie树还有种动态写法，使用指针和动态分配内存代替了连续的ch数组，更加节省内存。

Reference：http://blog.csdn.net/architect19/article/details/8966247

 

 

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
#define maxnode 400010
#define sigma_size 22

char a[maxnode][sigma_size];
int t[maxnode];
int n,m;
//struct Trie
//{
    int ch[maxnode][sigma_size];        //ch[i][j]：记录结点i的那个编号为j的子节点
    int val[maxnode];                    //val[i]：i节点的附加信息，
                                         //若val[i]=0不是还没到单词结束。否则val[i]为该单词的出现次数
    int sz;
    void Trie()
    {
        sz=1;
        memset(ch,0,sizeof(ch));
        memset(val,0,sizeof(val));
        memset(t,0,sizeof(t));
    }
    int idx(char c)                       //idx(c)即字符c的编号。
    {
        //A G C T
        if (c=='A')     return 1;
        if (c=='G')     return 2;
        if (c=='C')     return 3;
        if (c=='T')     return 4;
        //return c-'A';                    //第一个字符是A
    }
    void Insert(char s[sigma_size],int v)
    {
        int u=0,n=strlen(s);
        for (int i=0;i<n;i++)
        {
            int c=idx(s[i]);
            if (!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        //val[u]=v;
        val[u]+=v;
    }
    int Query(char s[sigma_size])     //times of s
    {
        int u=0,c;
        int tm=strlen(s);
        for (int i=0;i<tm;i++)
        {
            //c=s[i]-'A';
            c=idx(s[i]);
            if (!ch[u][c])    return 0;
            u=ch[u][c];
            //if (val[u]==1)  return true;    //若此时s还没走完但trie树上已经走到结尾了，即树上单词是s的前缀
        }
        return val[u];
    }
//};

int main()
{
    //freopen("in.txt","r",stdin);
    //ios::sync_with_stdio(false);

    //while (cin>>n>>m)
    while(~scanf("%d%d",&n,&m))
    {
        Trie();
        if ((n!=0)&&(m!=0))
        {
            for (int i=1;i<=n;i++)
            {
                //cin>>a[i];
                scanf("%s",a[i]);
                Insert(a[i],1);
            }
            for (int i=1;i<=n;i++)
            {
                int tm=Query(a[i]);
                t[tm]++;
            }
            for (int i=1;i<=n;i++)
                printf("%d\n",t[i]/i);
            //cout<<endl;
        }
        else break;
    }

    return 0;
}