详解Trie（c++）

Trie是一种高效的存储和查询字符串的数据结构

 

它主要用于字符串和文本的快速查找。Trie树是一种多叉树，它的每个节点都表示一个字符串的前缀，从根节点到叶子节点的路径组成了一个完整的字符串。Trie树的特点是时间复杂度为O(m)，其中m为待查字符串的长度，与数据规模无关，因此它非常适合于海量文本的高效匹配。Trie的实现方式有很多种，比如基于数组、链表、指针和位运算等。

Trie是一种特殊的数据结构，用于字符串和文本的快速查找。

今天就看一下利用trie来完成两道题目更好的理解和应用

维护一个字符串集合，支持两种操作：

1. I x 向集合中插入一个字符串 x；
2. Q x 询问一个字符串在集合中出现了多少次。

共有 N 个操作，所有输入的字符串总长度不超过 10^5，字符串仅包含小写英文字母。

输入如下

5
I abc
Q abc
Q ab
I ab
Q ab

输出如下

1
0
1

这个时候就可以用trie来储存和查询字符串

那trie是怎么储存的呢，一开始，以首字母为公共祖宗的子节点，然后依次储存接下来的每一个字符，最后，在该字符串的最后打一个标记。

那为什么要打一个标记呢，其实是为了更好的接下来的查找

比如例题中，一开始储存了abc，然后在abc中c打一个标记，第二次查询ab这个字符的时候，直接看该字符串的最后是不是c，显然不是，所以可以直接不用从头开始查询，减少时间复杂度

为了大家更好地理解，给出下面这张图片

 那接下来要怎么查找呢，首先查询在trie树里面查找该字符串和打了标记的结点，有的话再从该字符串头结点开始查找，一直遍历，看有没有一条路是不是和该字符串相等

接下来在完成一开始的问题，具体代码如下

#include<iostream>
#include<algorithm>

using namespace std;

const int N=100010;

int son[N][26],cnt[N],idx;
char str[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 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;
    cin>>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;
}

首先呢，该字符串只包含大小写字母，所以开一个son[N][26]，后面的26是因为一开始最多只有26条路

cnt[N]记录的是最后一个节点出现的次数

idx是记录当前路线，如果没有而且要执行插入的时候，就多开辟一条路线并且记录

下面是另外一道题目

在给定的 N 个整数 A1，A2……AN中选出两个进行 xor（异或）运算，得到的结果最大是多少？

输入格式

第一行输入一个整数 N。

第二行输入 N 个整数 A1～AN

输入数据

3
1 2 3

输出数据

3

如果是两个数异或结束以后再与剩下的一个数开始，那样的时间复杂度太高了

所以现在我们就可以利用trie来做这道题目

上图如果要求与2异或最大的路径就是5，因为要最大值，肯定要与自身当前二进制的位数情况相反。

所以我们就可以先储存每个数的二进制路线，然后要找最大值的时候可以直接查找

代码如下

#include<iostream>

using namespace std;

const int N=100010,M=3000010;

int son[M][2],idx;
int q[N];

void insert(int x)
{
    int p=0;
    for(int i=30;i>=0;i--)
    {
        int u=x>>i&1;
        if(!son[p][u]) son[p][u]=++idx;
        p=son[p][u];
    }
}

int query(int x)
{
    int p=0,res=0;
    for(int i=30;i>=0;i--)
    {
        int u=x>>i&1;
        if(son[p][!u])
        {
            p=son[p][!u];
            res=res*2+!u;
        }
        else 
        {
            p=son[p][u];
            res=res*2+u;
        }
    }
    return res;
}

int main()
{
    int n;
    scanf("%d",&n);
    
    for(int i=0;i<n;i++) scanf("%d",&q[i]);
    
    int res=0;
    
    for(int i=0;i<n;i++)
    {
        insert(q[i]);
        int t=query(q[i]);
        res=max(res,q[i]^t);
    }
    
    printf("%d",res);
    return 0;
}

这个时候一开始就有两条路径，所以son数组后面的为2