01字典树 HRBUST-1717

异或运算 (^)

参加运算的两个数,按二进制进行异或运算。
运算规则:0 ^ 0 = 0; 0 ^ 1 = 1; 1 ^ 0 = 0; 1 ^ 1 = 0;
(即:参加运算的两个数,对应位相“异”, 则该位为1, 否则为0)

01字典树

01字典树是一棵二叉树,最多32层
用于求解异或最值问题:给出一个数集合,给出一个数k,问集合中哪个数与k的异或值最大。

求解方法:要使得异或值最大,要尽量让对应位相“异”。
将集合中的数全部转换为二进制串,插入01字典树中。(从根节点到一个叶子结点,这条路径上边的值连起来得到一个二进制串。)
查询时,从根节点开始,每次查找与k的当前位相“异”的结点。

#define ll long long int 
ll tree[MAXN * 32][2];  // 01字典树,每个结点只有两个子节点
ll val[MAXN * 32];  // 结点的值, val[] = 0, 表示到当前结点还不能形成一个数,否则val[] = 数值
ll cnt;  // 结点编号
void init()
{
    memset(tree[0],0, sizeof(tree[0]));  // 初始化根节点
    cnt = 1; // 结点编号
}
void Insert(ll x)  // 插入一个数
{
    ll u = 0;
    for(int i = 31; i >= 0; i--)
    {
        ll c = (x>>i) & 1;  // 该数二进制的第i位是1还是0
        if(!tree[u][c])  // 结点未访问过
        {
            memset(tree[cnt], 0, sizeof(tree[cnt]));  // 初始化当前结点的边值
            val[cnt] = 0;  // 此时不能形成一个数,val[] = 0
            tree[u][c] = cnt++;  // 边结点指向结点的编号
        }
        u = tree[u][c];  // 下一个结点
    }
    val[u] = x;
}

ll query(ll x)  // 查询x与哪个数异或值最大
{
    ll u = 0;
    for(int i = 31; i >= 0; i--)
    {
        ll c = (x >> i) & 1;  // 获取二进制x的第i位
        if(tree[u][c^1]) u = tree[u][c^1];  // 取字典树中和当前位相“异”的位
        else u = tree[u][c];
    }
    return val[u];  // 到达叶子结点,返回叶子结点中的值
}

例题:HRBUST-1717

给出长为n的整数序列,截取这个序列的一个前缀和一个后缀,使得它们异或值最大(前后缀不可相交)

每次都把一个前缀的异或值加入01字典树中,(01字典树中所有的数看成是一个集合),
把这个前缀的后一位到最后一个数的异或值看成k,
那么问题就变成了,在集合中查找一个数,使得其与k的异或值最大。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
#define ll long long int
const int INF = 0x3f3f3f3f;
const int MAXN = 1e5 + 9;

int n;
ll a[MAXN], tree[MAXN * 32][2], val[MAXN * 32], cnt;
void init()
{
    memset(tree[0],0, sizeof(tree[0]));
    cnt = 1;
}
void Insert(ll x)
{
    ll u = 0;
    for(int i = 31; i >= 0; i--)
    {
        ll c = (x>>i) & 1;
        if(!tree[u][c])
        {
            memset(tree[cnt], 0, sizeof(tree[cnt]));
            val[cnt] = 0;
            tree[u][c] = cnt++;
        }
        u = tree[u][c];
    }
    val[u] = x;
}
ll query(ll x)
{
    ll u = 0;
    for(int i = 31; i >= 0; i--)
    {
        ll c = (x >> i) & 1;
        if(tree[u][c^1]) u = tree[u][c^1];
        else u = tree[u][c];
    }
    return val[u] ^ x;
}
int main()
{
    while(scanf("%d", &n) == 1)
    {
        init();
        ll ende = 0;   /// 全部异或值
        for(int i = 1; i <= n; i++)
        {
            scanf("%lld", &a[i]);
            ende ^= a[i];   
        }
        Insert(0);

        ll ans = 0, pre = 0;
        for(int i = 1; i <= n; i++)
        {
            pre ^= a[i];   /// 前i个数的异或值
            ende ^= a[i];  /// 再异或一下, 相当于没有与a[i]异或过,即得到了a[i + 1]到a[n]的异或值
            Insert(pre);   /// 加入这个前缀到字典树中
            ans = max(ans, query(ende)); /// 查找后缀与字典树中的最大异或值
        }
        printf("%lld\n", ans);
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值