字典树·Trie

Trie是一种高级的数据结构,用来统计多个字符串的查找,由于查找方式和内容和字典很想,所以被称为字典树

所以——只要你会查字典,你就会trie

来看一张图

在这里插入图片描述
我们现在这里有一个字典如上图,那么在字典里我们想把一个词给插进去,要怎么做呢,比如我们插入“清华园”,那么我们就先找“清”字,再找“华”,但是没有园,所以我们在华下面连到园

所以trie的插入和这个差不多,每次去找下一个字符,如果没有就建一个点

上代码:

inline void add(char s[]){
	int len=strlen(s);
	int u=1;
	for(int i=0;i<len;i++){
		int fin=s[i]-'a';
		if(!trie[u][fin])trie[u][fin]=++num;
		u=trie[u][fin];
	}
	end[u]++;
}

查询也很简单

	inline int ask(int* s){
		int u=1;
		int ans=0;
		Rep(i,1,s[0]){
			if(!trie[u][s[i]])return ans;
			u=trie[u][s[i]];
			ans+=end[u];
		}
		return ans-end[u]+cnt[u];
	}

因为每个题要查询的内容不太一样,所以trie的写法也不一样
对于这道题P2922 [USACO08DEC]秘密消息Secret Message

大概是这个样子的(结构体版)

struct dictionary_tree{
	int trie[N][2];
	int end[N],cnt[N];
	int sum;
	dictionary_tree(){
		memset(trie,0,sizeof(trie));
		memset(end,0,sizeof(end));
		memset(cnt,0,sizeof(end));
	}
	inline void add(int* s){
		int u=1;
		Rep(i,1,s[0]){
			if(!trie[u][s[i]])trie[u][s[i]]=++sum;
			u=trie[u][s[i]];
			cnt[u]++;
		}
		end[u]++;
	}
	inline int ask(int* s){
		int u=1;
		int ans=0;
		Rep(i,1,s[0]){
			if(!trie[u][s[i]])return ans;
			u=trie[u][s[i]];
			ans+=end[u];
		}
		return ans-end[u]+cnt[u];
	}
}Trie;
发布了18 篇原创文章 · 获赞 3 · 访问量 1263

猜你喜欢

转载自blog.csdn.net/devout_/article/details/100015832