可喜可贺。。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;
}