【POJ2001】Shortest Prefixes

Trie树

#include<cstdio>
#include<cstring>
using namespace std;
int trie[300010][30],sum[300010],cnt,n;
char s[1010][25];
void insert(int len,int k){
    int x=0;
    for (int i=1;i<=len;i++){
        if (!trie[x][s[k][i]-'a'])
            trie[x][s[k][i]-'a']=++cnt;
        x=trie[x][s[k][i]-'a'];sum[x]++;
    }
}
void ask(int len,int k){
    int x=0;
    for (int i=1;i<=len;i++){
        if (sum[x]==1) break;
        printf("%c",s[k][i]);
        x=trie[x][s[k][i]-'a'];
    }
}
int main(){
    cnt=n=0;
    while (scanf("%s",s[++n]+1)!=EOF)
        insert(strlen(s[n]+1),n);
    for (int i=1;i<=n;i++){
        printf("%s ",s[i]+1);
        ask(strlen(s[i]+1),i);
        printf("\n");
    }
    return 0;
}