pku2001 Shortest Prefixes

http://poj.org/problem?id=2001

数据结构，Trie

#include <stdio.h>
#include <string.h>

const int CHARSET = 26, BASE = 'a', MAX_NODE = 20100;

struct trie
{
    int tot, root, child[MAX_NODE][CHARSET];
    bool flag[MAX_NODE];
    int cnt[MAX_NODE];
    trie()
    {
        memset(child[1], 0, sizeof(child[1]));
        memset(cnt, 0, sizeof(cnt));
        flag[1] = false;
        root = tot = 1;
    }
    void insert(const char *str)
    {
        int *cur = &root;
        for(const char *p=str; *p; ++p)
        {
            cur = &child[*cur][*p-BASE];
            if(*cur == 0)
            {
                *cur = ++tot;
                memset(child[tot], 0, sizeof(child[tot]));
                flag[tot] = false;
            }
            cnt[*cur] ++;
        }
        flag[*cur] = true;
    }
    int query(const char *str)
    {
        int cur = root;
        int i = 1;
        int j;
        for(j=0; str[j]; j++, i++)
        {
            cur = child[cur][str[j]-BASE];
            if(cnt[cur] == 1)
            {
                return i;
            }
            if(str[j+1] == 0)
            {
                return i;
            }
        }
        return i;
    }
};
char s[1002][22]; 
int main()
{
    int i, j, n, m;
    memset(s, 0, sizeof(s));
    trie t;
    for(i=1; ~scanf("%s", s[i]); i++)
    {
        t.insert(s[i]);
    }
    n = i-1;
    for(i=1; i<=n; i++)
    {
        m = t.query(s[i]);
        printf("%s ", s[i]);
        for(j=0; j<m; j++)
        {
            printf("%c", s[i][j]);
        }
        printf("\n");
    }
    return 0;
}