Trie(算法版)

embedded/2025/1/20 2:28:16/
#include <iostream>using namespace std;const int N=100010;
int son[N][26],cnt[N],idx;
//son记录trie数,cnt记录每个词出现的次数,idx记录每个字符所占⽤的下标//加入字符串
void add(char str[]){//idx = 0既表⽰根节点也表⽰空节点int p =0;for(int i =0;str[i];i++){  //字符串末尾为0int u = str[i]-'a';if(!son[p][u]){son[p][u]= ++idx;//当前字符不存在,增加}p=son[p][u];  //向下继续遍历}cnt[p]++;  //p指向字符串最后一个字符,计数++
}int query(char str[]){int p =0;for(int i =0;str[i];i++){int u =str[i]-'a';if(!son[p][u]){return 0;//若没找到,直接返回0}p=son[p][u];}return cnt[p];  //p指向字符串的最后一个字符
}int main(){int n;cin>>n;while(n--){char op;char x[N];cin>>op>>x;if(op=='I')add(x);else cout<<query(x)<<endl;}return 0;}

假设情境 1096:我们有一个 Trie 树,开始时是空的。我们现在要插入以下两个单词:cat 和 car。

 

初始状态:

 
  • idx:初始值为 0,表示根节点。
  • son [N][26]:所有元素初始值为 0,表示 Trie 树是空的。
 

插入 “cat”:

 
  1. 字符 'c':
    • 计算索引 u = 'c' - 'a' = 2。
    • 检查 son [0][2],因为 Trie 树是空的,所以 son [0][2] 为 0。由于 son [0][2] 为 0,意味着当前节点 0 没有对应字符 'c' 的子节点。因此,我们创建一个新节点 1,并将 son [0][2] 设为 1,表示根节点 0 的 'c' 子节点是节点 1。
    • 然后,将 p 更新为 1,即移动到节点 1。
    • idx 增加到 1。
  2. 字符 'a':
    • 计算索引 u = 'a' - 'a' = 0。
    • 检查 son [1][0],此时 son [1][0] 仍为 0,因为之前还没有插入任何以 'a' 为第二个字符的单词。
    • 因此,我们创建一个新节点 2,并将 son [1][0] 设为 2。
    • 将 p 更新为 2,现在指向节点 2。
    • idx 增加到 2。
  3. 字符 't':
    • 计算索引 u = 't' - 'a' = 19。
    • 检查 son [2][19],此时 son [2][19] 为 0,因为还没有插入任何以 't' 为第三个字符的单词。
    • 创建新节点 3,并将 son [2][19] 设为 3。
    • p 更新为 3,现在指向节点 3。
    • idx 增加到 3。
    • 此时,已成功插入 “cat”。p 当前指向节点 3,在 cnt [3] 上加 1,表示有一个单词以 't' 结尾。
 

再插入 “car”:

 
  1. 字符 'c':
    • 计算索引 u = 'c' - 'a' = 2。
    • 检查 son [0][2],此时 son [0][2] 为 1,表示根节点已经有一个 'c' 子节点,指向节点 1。
    • 直接将 p 更新为 1,不用创建新节点。
  2. 字符 'a':
    • 计算索引 u = 'a' - 'a' = 0。
    • 检查 son [1][0],此时 son [1][0] 为 2,表示节点 1 有一个 'a' 子节点,指向节点 2。
    • 将 p 更新为 2。
  3. 字符 'r':
    • 计算索引 u = 'r' - 'a' = 17。
    • 检查 son [2][17],此时 son [2][17] 为 0,因为还没有插入任何以 'r' 为第三个字符的单词。
    • 创建新节点 4,并将 son [2][17] 设为 4。
    • p 更新为 4。
    • idx 增加到 4。
    • 现在已成功插入 “car”。p 当前指向节点 4,在 cnt [4] 上加 1,表示有一个单词以 'r' 结尾。
 

总结:
每当我们遇到一个新的字符且当前节点没有对应的子节点时(即 son [p][u] 为 0),我们就需要创建一个新的节点,并更新 son [p][u] 为新节点的编号。同时,idx 自增,用于分配新节点的编号。通过这种方式,Trie 树可以动态地构建,插入和查询操作都能高效完成。


http://www.ppmy.cn/embedded/155355.html

相关文章

智能问答系统中,思维链使用什么技术实现

智能问答系统中,思维链使用什么技术实现 目录 智能问答系统中,思维链使用什么技术实现基于规则的方法 技术原理:这种方法是通过预先定义一系列的规则和模板来构建思维链。开发者根据问题类型和领域知识,手动设计推理步骤的规则。例如,在数学运算问题中,规定先识别运算类型…

如何在idea中搭建SpringBoot项目

如何在idea中快速搭建SpringBoot项目 目录 如何在idea中快速搭建SpringBoot项目前言一、环境准备&#xff1a;搭建前的精心布局 1.下载jdk &#xff08;1&#xff09;安装JDK&#xff1a;&#xff08;2&#xff09;运行安装程序&#xff1a;&#xff08;3&#xff09;设置安装…

【C++笔记】红黑树

前言 各位读者朋友们大家好&#xff01;上期我们讲了二叉搜索树之一——AVL树&#xff0c;这一期我们继续讲解另一种二叉搜索树——红黑树。 目录 前言一. 红黑树的概念1.1 红黑树的规则1.2 红黑树如何确保最长路径不超过最短路径的两倍1.3 红黑树的效率 二. 红黑树的实现2.1…

Linux的常用命令(一)

目录 一、文件处理命令 1.文件处理命令ls 2.文件处理命令cd 3.文件处理命令pwd 4.文件处理命令touch 5.文件处理命令mkdir 6.文件处理命令cp 7.文件处理命令mv 8.文件处理命令rm 9.文件处理命令cat 10.文件处理命令more 11.文件处理命令head 12.文件处理命令tail …

大模型微调介绍-Prompt-Tuning

提示微调入门 NLP四范式 第一范式 基于「传统机器学习模型」的范式&#xff0c;如TF-IDF特征朴素贝叶斯等机器算法. 第二范式 基于「深度学习模型」的范式&#xff0c;如word2vec特征LSTM等深度学习算法&#xff0c;相比于第一范式&#xff0c;模型准确有所提高&#xff0c…

芝麻http/品易http/太阳http/极光http退市后,还有哪家好用推荐?

相信&#xff0c;已经有不少程序员朋友在讨论芝麻HTTP、品易HTTP、太阳HTTP和极光HTTP退市的消息。说实话&#xff0c;芝麻系HTTP代理服务商在代理IP圈子里可以说是有举足轻重的位置&#xff0c;曾经也是吸引了不少用户的青睐。2个月前它们的退市可以说让代理IP整个市场无论是用…

.NET 学习:从基础到进阶的全面指南

.NET学习资料 .NET学习资料 .NET学习资料 在当今软件开发的广阔领域中&#xff0c;.NET 是一个备受瞩目的开发平台&#xff0c;以其强大的功能、跨平台的特性以及丰富的生态系统&#xff0c;吸引着众多开发者投身其中。无论是构建企业级应用、Web 应用还是移动应用&#xff0…

【大数据2025】MapReduce

MapReduce 基础介绍 起源与发展&#xff1a;是 2004 年 10 月谷歌发表的 MAPREDUCE 论文的开源实现&#xff0c;最初用于大规模网页数据并行处理&#xff0c;现成为 Hadoop 核心子项目之一&#xff0c;是面向批处理的分布式计算框架。基本原理&#xff1a;分为 map 和 reduce …