POJ2001 - Shortest Prefixes - 字典树

#include<stdio.h>
#include<string.h>
char word[1001][21];
struct tree
{
	tree *next[30];
	int time;
	int fin;
	tree()
	{
		for(int i=0;i<30;i++)
		{
			next[i]=NULL;
		}
		time=0;
		fin=0;
	}
}*root;
void set(char *s)
{
	tree *item=root;
	for(int i=0;s[i];i++)
	{
		int j=s[i]-'a';
		if(item->next[j]==NULL)
		{
			item->next[j]=new tree;
			item->next[j]->time=1;
		}
		else
		{
			item->next[j]->time++;
		}
		item=item->next[j];
	}
	item->fin=1;
}
void search(char *s)
{
	tree *item=root;
	for(int i=0;s[i]&&item;i++)
	{
		int j=s[i]-'a';
		if(item->next[j]->time==1)
		{
			printf("%c",s[i]);
			return ;
		}
		else
		{
			printf("%c",s[i]);
		}
		item=item->next[j];
	}
}
int main()
{
	root=new tree;
	int i=0;
	while(gets(word[i]))
	{
		set(word[i]);
		i++;
	}
	int j=0;
	while(j<i)
	{
		printf("%s ",word[j]);
		search(word[j]);
		puts("");
		j++;
	}
}