poj 2001 Shortest Prefixes(Trie)

 

 

#include<cstdio>

#include<cstring>

#include<cmath>

#include<algorithm>

using namespace std;

int ch[30000][30];

int val[30000][30];

char str[1000+10][20+10];

int sz;

int idx(char c) {return c-'a';}

void insert(char *s)

{

    int u=0;

    int len=strlen(s);

    int i;

    for(i=0;i<len;i++)

    {

       int c=idx(s[i]);

       while(!ch[u][c])

       {

           memset(ch[sz],0,sizeof(sz));

           ch[u][c]=sz++;

       }

       val[u][c]++;

       u=ch[u][c];

    }

}

void print(char *s)

{

    int u=0;

    int len=strlen(s);

    int i,j,k;

    for(i=0;i<len;i++)

    {

        int c=idx(s[i]);

        printf("%c",s[i]);

        if(val[u][c]==1) return;

        u=ch[u][c];

    }

    return ;

}

int main()

{

    int n=0;

    int i,j,k;

    sz=1;

    memset(ch[0],0,sizeof(ch[0]));

    memset(val,0,sizeof(val));

    while(scanf("%s",str[n])!=EOF)

    {

        insert(str[n]);

        n++;

    }

    for(i=0;i<n;i++)

    {

        printf("%s ",str[i]);

        print(str[i]);

        printf("\n");

    }

    return 0;

}