字典树

本文介绍了如何使用字典树(Trie)解决字符串存储和查询问题,以及在最大异或对和牛异或问题中应用字典树进行优化。通过实例展示了如何插入字符串、查询字符串出现次数以及计算最大异或值。这些算法在处理大量字符串数据时能有效提高效率。
摘要由CSDN通过智能技术生成

字典树

#include <iostream>
#include <vector>
using namespace std;

const int N = 1e5 + 10;
int son[N][26], cnt[N], idx;//下标是0的点,即是空节点,又是根节点
char str[N];
void insert(char str[])//存储操作
{
    int p = 0;//初始化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];//将p值向下传递
    }
    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;
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n;
    cin >> n;
    while(n --)
    {
        char op[2];
        cin >> op >> str;
        if(op[0] == 'I') insert(str);
        else cout << query(str) << endl;
    }
}

最大异或对

在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;

const int N = 100010, M = 31 * N;
int a[N];
int son[M][2], idx;
int 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()
{

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

牛异或(前缀和,字典树)

在这里插入图片描述
输入:
5
1
0
5
4
2
输出:
6 4 5

#include <bits/stdc++.h>
using namespace std;

const int N = 100010, M = 21 * N;
int a[N];
int son[M][2], idx;
int n;
int id[M];
void insert(int x, int ix)
{
    int p = 0;
    for(int i = 20; i >= 0; i --)
    {
        int u = x >> i & 1;
        if(!son[p][u]) son[p][u] = ++ idx;
        p = son[p][u];
    }
    id[p] = ix;
}

int query(int x)
{
    int p = 0;
    for(int i = 20; i >= 0; i --)
    {
        int u = x >> i & 1;
        if(son[p][!u])
        {
            p = son[p][!u];
        }
        else{
            p = son[p][u];
        }
    }
    return id[p];
}

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i ++)
    {
        cin >> a[i];
        a[i] = a[i] ^ a[i - 1];
    }
    insert(a[0], 0);
    int res = -1, l, r;
    for(int i = 1; i <= n; i ++)
    {
        int k = query(a[i]);
        int t = a[i] ^ a[k];
        if (t > res) res = t, l = k + 1, r = i;
        insert(a[i], i);
    }
    cout << res << ' ' << l << ' ' << r << endl;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值