POJ 2001

可喜可贺。。poj 正常运转了。。

这道题依旧是字典树的题目

题干

输入 多个单词，找到能代表这个单词的前缀，最后将他们一起输出来，所谓能代表这个单词的前缀，就是与别的单词不同，举个例子， 几个单词 cartoon car cart 那么 cartoon 的前缀 只能是 carto 因为如果在短一些就会出现前缀重复了。

 

 

首先说一下思路，因为字典树这东西，记录了在第几个位置 26个字母中某个出现的次数，所以 我们既然要找最能代表某个单词的前缀和，那么就代表着 这个前缀最后一位字母 只在该位置出现过一次。有了这个思路，我们就可以开始改造一下字典树代码了。

开一个 sum int型 数组，来记录一下某个节点出现的次数，之后在 find 函数里，不断地去填补一个新的字符串，一直持续到循环结束或者， sum [ tree[ rt ] [ id ] ] == 1 再结束，之后这个填补好的字符串就是 我们的前缀了。

 

 

 

 

 

以下为 AC 代码

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 2e6+5;
int tree[maxn][30];
int tot;
int sum[maxn];
void _insert(char *str)
{
    int len = strlen(str);
    int rt = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - 'a';
        if(! tree[rt][id])
            tree[rt][id] = ++tot;
        sum[tree[rt][id]] ++;
        rt = tree[rt][id];
    }
}
string _find(char *str)
{
    int len = strlen(str);
    int rt = 0;
    string ans = "";
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - 'a';
        rt = tree[rt][id];
        ans += str[i];
        if(sum[rt] == 1)
            return ans;
    }
}
char str[1005][30];
int main()
{
    int tt= 0;
    while(scanf("%s" ,str[tt])!=EOF)
    {
        _insert(str[tt]);
        tt ++;
    }
    for(int i = 0; i < tt; i ++)
    {
        string tem = _find(str[i]);
        printf("%s %s\n" ,str[i] ,tem.c_str());
    }
    return 0;
}