POJ2001 Shortest Prefixes【字典树】

题目链接：

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

题目大意：

给一些字符串，求出每个字符串在这些字符串中能可以被唯一识别的最小前缀。

思路：

字典树的模板题，结构体中庸Count来存储字符前缀出现的次数，字典树中找到第一个

Count为1的前缀时，该前缀就是被唯一识别的最小前缀。

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

struct TrieNode
{
	int Count;
	struct TrieNode *Next[26];
};

TrieNode *root;

void Create()
{
	root = new TrieNode;
	memset(root->Next,NULL,sizeof(root->Next));
	root->Count = 0;
}

void Insert(char *s)
{
	TrieNode *p, *q;
	p = root;
	while(*s)
	{
		if(p->Next[*s-'a'] == NULL)
		{
			q = new TrieNode;
			memset(q->Next,NULL,sizeof(q->Next));
			q->Count = 1;
			p->Next[*s-'a'] = q;
		}
		else
			p->Next[*s-'a']->Count++;
		p = p->Next[*s-'a'];
		s++;
	}
}

void Find(char *s)
{
	TrieNode *p, *q;
	p = root;
	while(*s)
	{
		if(p->Count == 1)
			break;
		printf("%c",*s);
		p = p->Next[*s-'a'];
		s++;
	}
	printf("\n");
}

char word[1100][22];

int main()
{
	int k = 0;
	Create();
	while(scanf("%s",word[k]) != EOF)	
	{
		Insert(word[k]);
		k++;
	}
	for(int i = 0; i < k; ++i)
	{
		printf("%s ",word[i]);
		Find(word[i]);
	}
	
	return 0;
}