C++词典树/trie

模板题目

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

I x 向集合中插入一个字符串 x；
Q x 询问一个字符串在集合中出现了多少次。共有 N个操作，所有输入的字符串总长度不超过 1e5
，字符串仅包含小写英文字母。

输入格式
第一行包含整数 N，表示操作数。

接下来 N行，每行包含一个操作指令，指令为 I x 或 Q x 中的一种。

输出格式
对于每个询问指令 Q x，都要输出一个整数作为结果，表示 x在集合中出现的次数。每个结果占一行。

数据范围
1≤N≤2∗1e4
输入样例：
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例：
1
0
1

图解

代码

#include<iostream>

using namespace std;

const int N=1e5+10;//最长的行数为1e5；

int trie[N][26],index,cnt[N];//cnt是记录字符串出现次数的；index是创建新节点时的编号数，
//index也是让不同字符串结尾储存在trie中不同行的关键；

void insert(string s)
{
    int h=0;
    for(int i=0;i<s.size();i++)
    {
        int l=s[i]-'a';//将不同字母映射到数字上，组成一个个独特的列数l；
        if(trie[h][l]==0)
        {
            trie[h][l]=++index;
        }
        h=trie[h][l];
    }
    cnt[h]++;//每个字符串结尾元素所储存的行数h是独一无二的，因此能用cnt[h]来记录出现次数
}

int query(string s)
{
    int h=0;
    for(int i=0;i<s.size();i++)
    {
        int l=s[i]-'a';
        if(trie[h][l]==0)
        {
            return 0;
        }
        h=trie[h][l];
    }
    return cnt[h];
}

int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        char act;
        string s;
        cin>>act>>s;
        if(act=='I')
        {
            insert(s);
        }
        else if(act=='Q')
        {
            cout<<query(s)<<endl;
        }
    }
    return 0;
}

题目二

解题思路

什么是异或^：

注意事项：

cout << 2 ^ 3 ;//是错误代码，<<运算符优先级大于^，要加括号才能正确输出；
cout << (2 ^ 3);//正确代码

代码实现

#include<iostream>

using namespace std;

const int N = 31e5 + 10;

int trie[N][2], index, a[N];

void insert(int x)
{
    int h = 0;
    //将数字按二进制每一位存入trie树中
    for(int i = 30;i >= 0;i -- )
    {
        int l = (x >> i) % 2;//按31位算，第三十一位为符号位一定为0，所以从右移30位开始
        if(trie[h][l] == 0)
        {
            trie[h][l] = ++ index;
        }
        h = trie[h][l];
    }
}

int find_max_num(int x)//找出与x异或运算结果最大的数
{
    int h = 0, real_num = 0;
    for(int i = 30;i >= 0;i -- )
    {
        int l = ! ((x >> i) % 2);//每一位取反就是与x异或运算结果最大的数的二进制数；
        if(trie[h][l] == 0)
        {
            l = ! l ;
        }
        real_num = real_num * 2 + l;//二进制化为10进制
        h = trie[h][l];
    }
    return real_num;
}

int main()
{
    int n;
    cin >> n;
    for(int i = 0;i < n;i ++ )
    {
        scanf("%d", &a[i]);
        insert(a[i]);
    }
    
    int max_num = 0;
    
    for(int i = 0;i < n;i ++ )
    {
        int num = find_max_num(a[i]);
        if(max_num < (a[i] ^ num))
        {
            max_num = (a[i] ^ num);
        }
    }
    cout << max_num ;
    return 0;
}