左神算法学习日记——前缀树

前缀树的实现 

#include<iostream>
#include<string>
#include<vector>
#include<unordered_map>
#include<unordered_set>
using namespace std;

class node
{
public:
	int pass;
	int end;
	unordered_map<char,node*> edge;
	node() = default;
};

class tree
{
private:
	node* root;
public:
	tree()
	{
		root = new node();
	}
	void insert(string word)
	{
		node* temp = root;
		for (char var : word)
		{
			if (temp->edge.find(var) == temp->edge.end())//判断前缀树中是否有该便如果没有则添加该边
			{
				temp->edge.insert({ var,new node() });
			}
			temp = temp->edge[var];//移动到下一个节点
			temp->pass++;//记录有多少个单词有以该节点为终点的边所代表的字母
		}
		temp->end++;//每个以节点为终点的字母边都唯一锁定一个单词，end记录了前缀树中的该结点所锁定的单词出现的次数
	}
	int search(string word)
	{
		node* temp = root;
		for (char var : word)
		{
			if (temp->edge.find(var) == temp->edge.end())
			{
				return 0;
			}
			temp = temp->edge[var];
		}
		return temp->end;
	}
	void Delete(string word)
	{
		if (search(word))
		{
			node* temp = root;
			for (char var : word)
			{
				if (--temp->edge[var]->pass==0)
				{
					delete temp->edge[var];
					return;
				}
				temp = temp->edge[var];
			}
			--temp->end;
		}
	}
};