【Ybt OJ】[字符串算法第4章] 字典树Trie树

最新推荐文章于 2021-07-09 14:45:18 发布

EschatonAyane

最新推荐文章于 2021-07-09 14:45:18 发布

阅读量244

点赞数 1

分类专栏： YbtOJ 洛谷 trie树文章标签：字典树一本通

本文链接：https://blog.csdn.net/dgssl_xhy/article/details/113250417

版权

本文详细介绍了字典树（Trie树）在字符串算法中的应用，包括前缀统计、最大异或对、最长异或路径等例题的分析和解题思路。通过贪心策略证明了在某些问题中Trie树的优势，并提供了相关代码实现。

摘要由CSDN通过智能技术生成

$「$ 字符串算法 $」$ 第 $4$ 章字典树
目录：

A.前缀统计
B.最大异或对
C.最长异或路径
D.阅读理解

$A .$ 例题 $1$ 前缀统计

在这里插入图片描述

分析：

一道 $T r i e$ 树模板题需要支持插入 $(i n s e r t)$ 和查询 $(q u e r y)$ 两个操作就好了

CODE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue> 
#pragma GCC optimize(2)
#define reg register 
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int N=1e6+5;
struct Trie{
   
	int son[305],end;
	char ch;
}trie[N];
int n,m,lenx,p,tot,ret;
char x[N];
void insert(char a[])  //往字典树加入一个字符串
{
   
	lenx=strlen(a);
	p=0;
	for(int i=0;i<lenx;i++)
	{
   
		if(!trie[p].son[x[i]])
		{
   
			trie[p].son[x[i]]=++tot;
			trie[p].ch=x[i];
		}
		p=trie[p].son[x[i]];
	}
	trie[p].end++;
}
int search(char a[])  //查询
{
   
	lenx=strlen(a);
	p=0;ret=0;
	for(int i=0;i<lenx;i++)
	{
   
		if(!trie[p].son[x[i]]) return ret;
		p=trie[p].son[x[i]];
		ret+=trie[p].end;
	}
	return ret;
}
int main(){
   
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
   
		scanf("%s",x);
		insert(x);
	}
	for(int i=1;i<=m;i++)
	{
   	
		scanf("%s",x);
		printf("%d\n",search(x));
	}
	return 0;
}