字典树Trie总结

Trie就是把n个字符串建成一个树,树的每个节点都是一个字母,然后字符串的最后一个节点,我们会在Trie中对应的节点处打上标记,然后从根节点到每个标记点的路径上的字母依次排开就是n个字符串中的一个。

是不是很浅显?

稍稍举个例子:

n=5

he

his

hers

abcx

bc

那么我们建的Trie应该是这样的

          root

         /    |   \

      h      b    a

     /  \     |      \

  *e    i    *c      b

  |      |              |

  r     *s            c

  |                    |

  *s                 *x


其中打*的就是标记点,即一个单词的结尾。

至于建Trie,就弄个链式前向星就好了,

设ch[u][c]表示u号节点的字母ASCII码值-1为c的孩子的编号,所以0<=c<26

那么在上面的例子中,h点的编号为1,e点的编号为4,所以ch[1][4]=4,

插入时就一个字母依次的,如果当前ch[u][c]为空,就新设一个点,最后在u打上标记。

查找时也是一个字母依次的,如果当前ch[u][c]为空,查找失败,否则把u调到ch[u][c],如果最后一个字母也找到了,再看看当前点是否有标记,没有标记则查找失败。反之,则查找成功。

如果没怎么看懂,就结合代码看看,其实很容易理解的。

以洛谷P2292,HNOI2004L语言为例。

一个Trie+很水的dp,时限卡的有点紧。

参考代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e6+10;
int n,m,len=0;
char P[N];
bool dp[N];
struct Trie{
	int ch[211][26],val[211],NodeCnt;
	inline void clear(){
		memset(ch,0,sizeof(ch));
		memset(val,0,sizeof(val));
		NodeCnt=0;
	}
	inline void insert(char *s){
		int u=0,l=strlen(s);
		len=max(len,l);
		for (int i=0;i<l;i++){
			int c=s[i]-'a';
			if (!ch[u][c])ch[u][c]=++NodeCnt;
			u=ch[u][c];
		}
		val[u]=1;
	}
	inline bool find(int st,int en){
		int u=0;
		for (int i=st;i<=en;i++){
			int c=P[i]-'a';
			if (!ch[u][c])return 0;
			u=ch[u][c];
		}
		return val[u];
	}
}T;
int main(){
	scanf("%d%d",&n,&m);
	T.clear();
	for (int i=1;i<=n;i++){
		scanf("%s",P);
		T.insert(P);
	}
	for (int i=1;i<=m;i++){
		scanf("%s",P);
		memset(dp,0,sizeof(dp));
		int l=strlen(P),ans=0;
		for (int j=0;j<l;j++)
			for (int k=max(j-len,-1);k<=j;k++)
				if ((k==-1 || dp[k]) && T.find(k+1,j)){
					dp[j]=1;ans=j+1;break;
				}
		printf("%d\n",ans);
	}
	return 0;
}


裸的Trie题并不多,常常是做数据结构优化dp,贪心什么的,并不是核心。

    原文作者:Trie树
    原文地址: https://blog.csdn.net/no1_terminator/article/details/77686813
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞