Tire树的学习

本文参照http://blog.csdn.net/niushuai666/article/details/7002823，写的很详细。大体记住的是：插入节点。构建失败指针，查询散步操作。

下面代码中的一个缺陷是没有对堆空间进行释放。

#include <iostream>  
#include<string>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;
const int N=26;
struct Node{
	Node():count(0),fail(NULL){
	  memset(next,NULL,sizeof(next));
	}
	int count;
	Node* fail;
	Node* next[N];
};
void insert(Node* root,const string& str){
	for(size_t i=0;i<str.size();++i){
	  int ind=str[i]-'a';
	  if(root->next[ind]==NULL)
		  root->next[ind]=new Node();
	  root=root->next[ind];
	}
	root->count=1;
}
void buildFail(Node* root){
	queue<Node* > que;
	que.push(root);
	Node* tmp,*p;
	while(!que.empty()){
		tmp=que.front();
		que.pop();
		for(int i=0;i<N;++i){
			if(tmp->next[i]==NULL) continue;
			if(tmp==root) tmp->next[i]->fail=root;
			else{
				p=tmp->fail;
				while(p!=NULL){
					if(p->next[i]){
						tmp->next[i]->fail=p->next[i];
						break;
					}
					p=p->fail;
				}
				if(p==NULL) tmp->next[i]->fail=root;
			}
			que.push(tmp->next[i]);
		}
	}
}
int query(Node* root,const string& query){
	Node* p=root;
	int cnt=0;
	for(int i=0;i<query.size();++i){
	   int ind=query[i]-'a';
	   //匹配失败并且不是根，所以从fail指针开始匹配
	   while(p->next[ind]==NULL&&p!=root) p=p->fail;
	   p=p->next[ind];
	   if(p==NULL) p=root;
	   Node* tmp=p;
	   while(tmp!=root&&tmp->count!=-1){
		   cnt+=tmp->count;
		   tmp->count=-1;
		   tmp=tmp->fail;
	   }
	}
	return cnt;
}
int main()  
{  
	string strs[]={"say","she","shr","he","her"};
	Node * root=new Node;
	for(int i=0;i<5;++i){
	  insert(root,strs[i]);
	}
	buildFail(root);
	cout<<query(root,"yasherhs")<<endl;

	 system("pause");
        return 0;     
}