Trie树

(2012-07-20 20:30:16)

自己写了一个Trie树，以后当模板吧。不过临时写也不复杂。

#include<iostream>
#include<cstring>
#define kind 26//26个小写字母。
using namespace std;
struct node
{
	bool is_word;//表示是否为一个单词，是则为true，否为false。
	node *son[kind];//指向子结点的指针数组。
	node()//构造函数。
	{
		memset(son,NULL,sizeof(son));
	}
};
void insert(node *root,char *pattern)//插入一个单词。
{
	if(root==NULL)
		return ;
	int i;
	for(i=0;i<strlen(pattern);i++)
	{
		if(root->son[pattern[i]-'a'])
		{
			root=root->son[pattern[i]-'a'];
			continue;
		}
		node *temp=new node();///
		if(i==strlen(pattern)-1)
			temp->is_word=true;
		else
			temp->is_word=false;
		root->son[pattern[i]-'a']=temp;
		root=temp;
	}
}
bool search(node *root,char *pattern)//搜索一个单词。
{
	int i;
	for(i=0;i<strlen(pattern);i++)
	{
		if(root==NULL)
			break;
		root=root->son[pattern[i]-'a'];
	}
	if(root && root->is_word)
		return true;
	return false;
}
void del(node *root)//删除整个Trie树。
{
	int i;
	for(i=0;i<kind;i++)
	{
		if(root->son[i])
			del(root->son[i]);
	}
	delete root;
}
int main()
{
	int test;
	char s[105];
	scanf("%d",&test);
	while(test--)
	{
		int m,n;
		scanf("%d",&m);
		node *root=new node();
		root->is_word=false;
		int i;
		for(i=0;i<m;i++)
		{
			scanf("%s",s);
			insert(root,s);
		}
		scanf("%d",&n);
		for(i=0;i<n;i++)
		{
			scanf("%s",s);
			if(search(root,s))
				cout<<"Yes!"<<endl;
			else
				cout<<"No!"<<endl;
		}
		del(root);
	}
	return 0;
}

补充：
(1)
node *p=new node();
(2)
node *p;
node temp;
temp.is_word=false;
p=&temp;
(1)(2)的效果是一样的，但建议使用(1)方法。因为在循环中(2)所产生的p值，也即是temp的地址全是相同的!