CF1851F 题解

题意:

给定 n 个数的序列 a,和一个数 k,求 i,j[1,n],=ijx[0,2k],使得 (aix)&(ajx) 最大,其中 为异或符号,& 为按位与符号。

01-trie 好题。

考虑已经插入 a1ap1,如何找到 j=p 时使答案最大的 ix

显然可以从二进制高位向低位贪心。

考虑单独的二进制位对答案的影响:

ai aj x (aix)&(ajx)
1 1 0 1
0 0 1 1
0 1 0/1 0
1 0 0/1 0

(上图表列出了使 (aix)&(ajx) 最大的 x 取值)

若当前位相同,则可以通过改变 x 值使结果位为 1

若当前位不同,则无论 x 值为多少,结果位均为 0

于是发现在 trie 上每位贪心地选与当前位相同的数即可。

代码:

#include<bits/stdc++.h>
#define ll long long
#define MAXN 200010
using namespace std;

ll read(){
    char ch=getchar();
    ll s=0,w=1;
    while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
    return s*w;
}

ll n, k;

struct trie{
    ll son[2];
    ll pos;
}tree[MAXN << 6];
ll now = 0;

ll a[MAXN];
ll ans = 0, ansx = 0, anspos1 = 2, anspos2 = 1;

void query(ll x, ll num, ll bit, ll cnt, ll cntt, ll pos){
    if(bit == -1){
        if(ans < cnt){
            ans = cnt;
            ansx = cntt;
            anspos1 = pos;
            anspos2 = tree[x].pos;
        }
        return;
    }
    bool p = num & (1 << bit);
    if(tree[x].son[p]){
        cnt ^= (1 << bit);
        if(p == 0){
            cntt ^= (1 << bit);
        }
        query(tree[x].son[p], num, bit-1, cnt, cntt, pos);
    }else{
        query(tree[x].son[p^1], num, bit-1, cnt, cntt, pos);
    }
}

void insert(ll x, ll num, ll bit, ll pos){
    if(bit == -1){
        tree[x].pos = pos;
        return;
    }
    bool p = num & (1 << bit);
    if(!tree[x].son[p]){
        tree[x].son[p] = ++now;
    }
    insert(tree[x].son[p], num, bit-1, pos);
}

int main(){
    ll T = read();

    while(T--){
        now = 0, ans = 0;
        anspos1 = 2, anspos2 = 1;
        ansx = 0;

        n = read(), k = read();

        for(ll i = 0; i <= n * k; i++){
            tree[i].pos = 0;
            tree[i].son[0] = 0, tree[i].son[1] = 0;
        }

        for(ll i = 1; i <= n; i++){
            a[i] = read();
            query(0, a[i], k-1, 0, 0, i);
            insert(0, a[i], k-1, i);
        }

        cout << anspos2 << " " << anspos1 << " " << ansx << endl;
    }

    return 0;
}
posted @   象征阳光  阅读(5)  评论(0编辑  收藏  举报  
编辑推荐:
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 使用C#创建一个MCP客户端
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
点击右上角即可分享
微信分享提示