关闭

hdu 1671|| zoj 2876 phone list(trie 水)

712人阅读 评论(0) 收藏 举报

唉,难道说在大牛的肩膀上做题会容易么-,-

2826252 2012-03-28 00:36:32 Accepted 2876 C++ 250 2688 zisu_123


之前那题不需要加上dealTrie释放空间,这题不行,不然MLE的。

不过加上总是一个好习惯呗。

模版-,-Orz

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define MAX 10
typedef struct Trie{
	Trie *next[MAX];
	int n;
}Trie;
Trie *root;
void createTrie(char *str)
{
	int len=strlen(str);
	Trie *p=root,*q;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'0';
		if(p->next[id]==NULL)
		{
			q=(Trie *)malloc(sizeof(Trie));
			q->n=1;
			for(int j=0;j<MAX;j++)
				q->next[j]=NULL;
			p->next[id]=q;
			p=p->next[id];
		}
		else
		{
			p->next[id]->n++;
			p=p->next[id];
		}
	}
	p->n=-1;
}

int findTrie(char *str)
{
	int len=strlen(str);
	Trie *p=root;
	for(int i=0;i<len;i++)
	{
		int id=str[i]-'0';
		p=p->next[id];
		if(p==NULL) return 0;
		if(p->n==-1)
			return -1;

	}
	return -1;
}
int dealTrie(Trie* T)
{
	int i;
	if(T==NULL)
		return 0;
	for(i=0;i<MAX;i++)
		if(T->next[i]!=NULL)
			dealTrie(T->next[i]);
	free(T);
	return 0;
}
int main()
{
	int n,m;
	char num[11];
	scanf("%d",&n);
	while(n--)
	{
		int flag=0;
		root=(Trie *)malloc(sizeof(Trie));
		for(int i=0;i<MAX;i++)
			root->next[i]=NULL;
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			scanf("%s",num);
			if(findTrie(num)==-1) flag=1;
			createTrie(num);
		}
		if(flag) printf("NO\n");
		else printf("YES\n");
		dealTrie(root);
	}
	return 0;
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:384532次
    • 积分:8282
    • 等级:
    • 排名:第2378名
    • 原创:436篇
    • 转载:24篇
    • 译文:0篇
    • 评论:29条
    文章分类
    最新评论
    even's code life