POJ 2001 Shortest Prefixes(字典树Trie)

POJ 2001 Shortest Prefixes(字典树Trie)

http://poj.org/problem?id=2001

题意:

        给你多个单词组成的词典，现在要你输出每个单词对应的前缀，使得每个前缀唯一。如果不存在唯一前缀就直接输出单词本身。比如字典里面有car 和caraaab 两个单词，那么对于caraaab就输出cara，对于car就输出car。

分析:

       直接用所有单词建立字典树,并且字典树节点的v值表示以该节点为根的子树下有多少个单词.

       然后对于每个单词查找一遍字典树,顺着当前单词的字母往下走,如果走要一个节点的v值==1,那么就直接输出该节点并返回,否则输出当前的完整单词.

AC代码:1A

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
using namespace std;
#define maxn 26;
const int maxenode=20000+100;
const int maxnsize=26;


struct Trie
{
	int ch[maxenode][maxnsize];
	int val[maxenode];
	int sz;
	void clear()
	{
		sz=1;
		memset(ch,0,sizeof ch);
		memset(val,0,sizeof val);
	}
	int idx(char x)
	{
		return x-'a';
	}
	void insert (char *s)
	{
		int u =0,n=strlen(s);
		for(int i=0;i<n;i++)
		{
			int id=idx(s[i]);
			if(ch[u][id]==0)
			{
				ch[u][id]=sz;
				memset(ch[sz],0,sizeof ch[sz]);
				sz++;
			}
			u=ch[u][id];
			val[u]++;
		}
	}
	int  find(char *s)
	{
		int u=0;
		int len=strlen(s);
		for(int i=0;i<len;i++)
		{
			int id=idx(s[i]);

			u=ch[u][id];
			if(val[u]==1)
				return i;
		}
		return len-1;
	}
}trie;
char s[1100][21];
int main()
{
	trie.clear();
	int tot=0;
	while(scanf("%s",s[tot])!=EOF)
	{
		trie.insert(s[tot]);
		tot++;
	}
	//cout<<tot<<endl;
	for(int i=0;i<tot;i++)
	{
		int n=trie.find(s[i]);
		printf("%s ",s[i]);
		for(int j=0;j<=n;j++)
			printf("%c",s[i][j]);
		printf("\n");
	}
}