POJ 2001 Shortest Prefixes(字典树Trie)

POJ 2001 Shortest Prefixes(字典树Trie)

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

题意:

        给你多个单词组成的词典，现在要你输出每个单词对应的前缀，使得每个前缀唯一。如果不存在唯一前缀就直接输出单词本身。比如字典里面有car 和caraaab 两个单词，那么对于caraaab就输出cara，对于car就输出car。

分析:

       直接用所有单词建立字典树,并且字典树节点的v值表示以该节点为根的子树下有多少个单词.

       然后对于每个单词查找一遍字典树,顺着当前单词的字母往下走,如果走要一个节点的v值==1,那么就直接输出该节点并返回,否则输出当前的完整单词.

AC代码:1A

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int maxnode=20000+100;
const int sigma_size=26;
struct Trie
{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;
    void init()
    {
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));
        val[0]=0;
    }
    void insert(char *s)
    {
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            if(ch[u][id]==0)
            {
                ch[u][id]=sz;
                val[sz]=0;
                memset(ch[sz],0,sizeof(ch[sz]));
                sz++;
            }
            u=ch[u][id];
            val[u]++;
        }
    }
    void find(char *s)
    {
        int n=strlen(s),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            u=ch[u][id];
            printf("%c",s[i]);
            if(val[u]==1)
                return ;
        }
    }
}trie;
char word[1000][25];
int cnt;
int main()
{
    cnt=0;
    trie.init();
    while(scanf("%s",word[cnt])==1)
    {
        trie.insert(word[cnt++]);
    }
    for(int i=0;i<cnt;i++)
    {
        printf("%s ",word[i]);
        trie.find(word[i]);
        printf("\n");
    }
    return 0;
}