【ybtoj】【字典树】最短前缀

题意

F. 2.最短前缀

内存限制：256 MiB时间限制：1000 ms标准输入输出

题目类型：传统评测方式：文本比较

提交提交记录返回课程章节

题目描述

字符串的前缀是指以字符串的开头开始的子字符串。

如 carbon 的前缀有：c、ca、car、carb、carbo、carbon。

给定一组单词，求每个单词独一无二的最短前缀。

若找不到，则这个单词独一无二的最短前缀为它本身。

输入格式

输入包含若干行，每行包含一个字符串 。

输出格式

对于每个单词，输出一行两个字符串，分别表示输入对应行的单词以及该单词独一无二的最短前缀，两个字符串用空格隔开。

样例

样例输入

carbohydrate
cart
carburetor
caramel
caribou
carbonic
cartilage
carbon
carriage
carton
car
carbonate

样例输出

carbohydrate carboh
cart cart
carburetor carbu
caramel cara
caribou cari
carbonic carboni
cartilage carti
carbon carbon
carriage carr
carton carto
car car
carbonate carbona

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int INF = 0x3f3f3f3f,N = 1e3+10;
int trie[N*20][30],tot=1,cnt,vis[N*20];
char s[N][22]; 
void insert(int id)
{
    int p=1,len=strlen(s[id]+1);
    for(int i=1;i<=len;i++)
    {
        int ch=s[id][i]-'a';
        if(!trie[p][ch]) trie[p][ch]=++tot;
        p=trie[p][ch];
        vis[p]++;
    }
}
int find(int id)
{
    int p=1,len=strlen(s[id]+1);
    for(int i=1;i<=len;i++)
    {
        int ch=s[id][i]-'a';
        p=trie[p][ch];
        if(vis[p]==1) return i; 
    }
    return len;
}
void print()
{
    for(int i=1;i<=cnt;i++)
    {
        int len=find(i);
        printf("%s ",s[i]+1);
        for(int j=1;j<=len;j++)    printf("%c",s[i][j]);
        printf("
");
    }
}
int main()
{    
    while(scanf("%s",s[++cnt]+1)!=EOF) insert(cnt);
    print();
    return 0;
}