POJ 3630 Phone List(Trie树，静态数组实现)

/*
这道题动态分配内存会超时
先建图，建图完成后，再判断，这样不容易出错
*/


/*
解法一：
Trie树，静态数组实现
*/

#include <cstdio>
#include <cstring>
const int nMax = 200000;
struct Trie
{
	Trie *next[10];
	int count;
}trie[nMax];
int pos;

void buildTrie(char *s, int len)
{
	Trie *p = &trie[0];
	int i;
	for(i = 0; i < len; ++ i)
	{
		int index = s[i] - '0';
		if(p->next[index] == NULL) p->next[index] = &trie[pos ++];//模拟动态分配内存
		p = p->next[index];
	}
	p->count ++;
}

bool judge(Trie *root)//学会使用函数的返回值，减少全局变量的使用有时候可以简化程序复杂度
{
	int i;
	if(root->count > 1) return 1;
	for(i = 0; i < 10; ++ i)
	{
		if(root->next[i])
		{
			if(root->count > 0) return 1;
			if(judge(root->next[i])) return 1;//原来这里没有做处理
		}
	}
	return 0;
}

int main()
{
	//freopen("e://data.in", "r", stdin);
	int T;
	scanf("%d", &T);
	while(T --)
	{
		int N;
		scanf("%d", &N);
		int i;
		char s[15];
		pos = 1;
		memset(trie, 0, sizeof(trie));//memset可以对数组进行初始化
		for(i = 0; i < N; ++ i)
		{
			scanf("%s", s);
			int len = strlen(s);
			buildTrie(s, len);
		}
		printf("%s\n", judge(&trie[0]) ? "NO" : "YES");
	}
	return 0;
}

/*
解法二：哈希表实现
*/

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int HASH = 12007;
int T, N;
char s[10007][15];
int flag;
char hash[HASH][15];
int head[HASH];

int fhash(char *s, int len)
{	
	int p = 0;
	int i;
	for(i = 0; i < len; ++ i)
		p = (p * 10 + s[i] - '0') % HASH;
	return p;
}

int insert(char *s, int len)
{
	int p = fhash(s, len);
	while(head[p] != 0 && strcmp(s, hash[p]) != 0)
	{
		p = (p + 1) % HASH;
	}
	if(head[p] == 0)
	{
		strcpy(hash[p], s);
		head[p] = 1;
		return 1;
	}
	return 0;
}

bool isFind(char *s, int len)
{
	char ss[15];
	int i;
	for(i = 0; i < len; ++ i)
		ss[i] = s[i];
	ss[len] = '\0';
	int p = fhash(ss, len);
	while(head[p] != 0 && strcmp(ss, hash[p]) != 0)
	{
		p = (p + 1) % HASH;
	}
	if(head[p] == 0)
	{
		return 0;
	}
	return 1;
}

int main()
{
	//freopen("e://data.in", "r", stdin);
	int T;
	scanf("%d", &T);
	while(T --)
	{
		memset(head, 0, sizeof(head));
		scanf("%d", &N);
		int i, j;
		int len;
		flag = 0;
		for(i = 0; i < N; ++ i)
		{
			scanf("%s", s[i]);
			if(flag == 0)
			{
				len = strlen(s[i]);
				if(!insert(s[i], len))
				{
					flag = 1;
				}
			}
		}
		for(i = 0; i < N && !flag; ++ i)
		{
			len = strlen(s[i]);
			for(j = 1; j < len && !flag; ++ j)
				if(isFind(s[i], j))
					flag = 1;
		}
		if(flag)
			printf("NO\n");
		else
			printf("YES\n");
	}
	return 0;
}