hdu 1251 统计难题 trie

#include<bits/stdc++.h>
using namespace std;
int ch[500000][26],sz,val[500000];

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

void trie(char *s)
{
    int u=0,c,i,len=strlen(s);
    for(i=0;i<len;i++)
    {
        c=idx(s[i]);
        if(ch[u][c]==0)
        {
            ch[u][c]=sz;
            sz++;
        }
        u=ch[u][c];
        val[u]++;
    }
}

int main()
{
    char s[20];
    int ans,i,len,u,c;
    sz=1;
    memset(val,0,sizeof(val));
    memset(ch,0,sizeof(ch));
    while(gets(s))
    {
        if(strlen(s)==0) break;
        trie(s);
    }
    while(gets(s))
    {
        ans=0;
        len=strlen(s);
        u=0;
        for(i=0;i<len;i++)
        {
            c=idx(s[i]);
            if(ch[u][c]==0) break;
            u=ch[u][c];
        }
        if(i==len) ans=val[u];
        printf("%d\n",ans);
    }
    return 0;
}