AcWing算法基础课笔记——trie树

AcWing算法基础课第二章数据结构中有关Trie树的笔记。
欢迎访问个人网站https://hjh645.github.io/
(>w<)~

Trie 树

快速存储和查找字符串集合的数据结构.

模板题

#include <iostream>
using namespace std;
const int N = 100010;

// son[N][26]记录结点的下一个子节点的位置
// cnt[N]记录当前符号的个数
// idx表示数组的末尾
int son[N][26], cnt[N], idx;  

char str[N];
    
// 插入字符串
void insert(char str[]) {
    int p = 0; //从根节点开始找
    for(int i = 0; str[i]; i ++ ) { // '\0'是false
        int u = str[i] - 'a';
        if(!son[p][u]) son[p][u] = ++ idx; // 如果p结点的下一个u字母不存在，++idx
        p = son[p][u]
    }
    cnt[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;
        p = son[p][u];
    }
    return cnt[p];
}

int main() {
    int n;
    scanf("%d", &n);
    while(n --) {
        char op[2];
        scanf("%s%s", op, str);
        if(op[0] == 'I') insert(str)
        else printf("%d\n", query(str));
    }
    
    return 0;
}

练习1.1 P2580 于是他错误的点名开始了

来自洛谷题单：Trie 树 基础练习 - 题单 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

该题传送门：P2580 于是他错误的点名开始了 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目背景

XS中学化学竞赛组教练是一个酷爱炉石的人。

他会一边搓炉石一边点名以至于有一天他连续点到了某个同学两次，然后正好被路过的校长发现了然后就是一顿欧拉欧拉欧拉（详情请见已结束比赛 CON900）。

题目描述

这之后校长任命你为特派探员，每天记录他的点名。校长会提供化学竞赛学生的人数和名单，而你需要告诉校长他有没有点错名。（为什么不直接不让他玩炉石。）

输入格式

第一行一个整数 $n$，表示班上人数。

接下来 $n$ 行，每行一个字符串表示其名字（互不相同，且只含小写字母，长度不超过 $50$）。

第 $n+2$ 行一个整数 $m$，表示教练报的名字个数。

接下来 $m$ 行，每行一个字符串表示教练报的名字（只含小写字母，且长度不超过 $50$）。

输出格式

对于每个教练报的名字，输出一行。

如果该名字正确且是第一次出现，输出 OK，如果该名字错误，输出 WRONG，如果该名字正确但不是第一次出现，输出 REPEAT。

样例 #1

样例输入 #1

5  
a
b
c
ad
acd
3
a
a
e

样例输出 #1

OK
REPEAT
WRONG

提示

• 对于 $40\%$ 的数据，$n\le 1000$，$m\le 2000$。
• 对于 $70\%$ 的数据，$n\le 10^4$，$m\le 2\times 10^4$。
• 对于 $100\%$ 的数据，$n\le 10^4$，$m\le 10^5$。

---

$\text{upd 2022.7.30}$：新增加一组 Hack 数据。

我的解答

我的解答是直接套用模板：

#include<iostream>
using namespace std;

const int N = 2000010;
char str[N];
int son[N][26], cnt[N], idx, q[N];

void insert(char str[]) {
    int p = 0;
    for(int i = 0; str[i]; i ++ ) {
        int u = str[i] - 'a';
        if(!son[p][u]) son[p][u] = ++ idx;
        p = son[p][u];
    }
    cnt[p] ++;
}

int qurey(char str[]) {
    int p = 0;
    for(int i = 0; str[i]; i++ ) {
        int u = str[i] - 'a';
        if(!son[p][u]) return -1;  // 返回-1代表错误
        p = son[p][u];
    }
    if(cnt[p] == 0) return -1;
    q[p] ++;
    if(q[p] > 1) return 1; //表示重复 
	return 0; //表示正确
}

int main() {
    int n, m;
    scanf("%d", &n);
    for(int i = 0; i < n; i ++) {
        scanf("%s", str);
        insert(str);
    }
    scanf("%d", &m);
    for(int i = 0; i < m; i ++) {
        scanf("%s", str);
        int ans = qurey(str);
        if(ans == 0) printf("OK\n");
        else if (ans == -1) printf("WRONG\n");
        else printf("REPEAT\n");
    }
    return 0;
}