HDU 1671 Phone List(字典树Trie)

HDU 1671 Phone List(字典树Trie)

http://acm.hdu.edu.cn/showproblem.php?pid=1671

题意:

        给你多个由0-9构成的字符串集合,问你这个集合中是否有一个字符串是其他字符串的前缀?

分析:

       直接构造字典树,然后一一插入每个字符串,判断是否有比当前字符串短或长的同一路径的字符串已经在树中了.

       怎么判断？就是对于每个字符串路径上的结点加1，对于每个字符串的结束结点加12000（大于n的一个数即可）

当在查找一个字符串路径时，如果路径上有一个结点大于等于12000，说明有比当前字符串短的同一路径的字符串已经在树中了：如果当查找到字符串最后一个字符的结点的值大于0的话，说明有比当前字符串长的同一路径的字符串已经在树中了；

AC代码:

#include<cstdio>
#include<cstring>
#include<vector>
#include<iostream>
using namespace std;
#define maxn 26;
const int maxenode=100000+100;
const int maxnsize=26;
const int over=12000;
bool flag;
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-'0';
	}
	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];
			if(i!=n-1)
			{
				if(val[u]>=over)
					flag=false;
				val[u]++;
			}
		}
		if(val[u]!=0)flag=false;
		val[u]+=over;//单词结点记录单词的长度
	}
	void find(char *s)
	{
		int u=0;
		int len=strlen(s);
		for(int i=0;i<len;i++)
		{
			int id=idx(s[i]);
			if(ch[u][id]==0)
				return ;
			u=ch[u][id];
		}
	}
}trie;

int main()
{
	int t;
	scanf("%d",&t);
	int cas=1;
	char s[20];
	while(t--)
	{
		int n;
		scanf("%d",&n);
		flag=1;
		trie.clear();
		for(int i=0;i<n;i++)
		{
			scanf("%s",s);
			//cout<<"++";
			if(flag)
			{
				trie.insert(s);
			}
		}
		if(flag)printf("YES\n");
		else printf("NO\n");
	}
}