hdu 5536(trie + 贪心)

最新推荐文章于 2021-07-23 14:32:50 发布

路小白_zZ

最新推荐文章于 2021-07-23 14:32:50 发布

阅读量506

点赞数

分类专栏： ACM-字典树

本文链接：https://blog.csdn.net/hyczms/article/details/49641953

版权

ACM-字典树专栏收录该内容

11 篇文章 0 订阅

订阅专栏

题意：有n个数字，选择三个下标不同的数字，其中两个求和，然后和第三个数字异或得到的最后结果是A，问A最大是多少。
题解：长春赛区的一题，据说可以三层for过掉，还是用字典树好好写一遍，首先枚举把n个数字按二进制码插入字典树，维护val[u]节点u是多少个数字的前缀，id[u]是节点u是数字id[u]的最后一个节点，然后枚举两个求和的点，把这两个点在字典树上用flag[u]标记起来，然后把sum = a[i] + a[j]拿到字典树去贪心，如果当前位是0，就向1走，反之亦然，如果反向走到被标记的唯一单支路径(或者有两条路径但都是被标记的)，就走和当前位相同的节点。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 1000 * 33;
int Next[N][2], val[N], flag[N], sz, n;
int id[N], a[1005];

void init() {
    memset(Next[0], 0, sizeof(Next[0]));
    val[0] = flag[0] = id[0] = 0;
    sz = 1;
}

void insert(int s) {
    int u = 0;
    for (int i = 31; i >= 0; i--) {
        int k = (s >> i) & 1;
        if (!Next[u][k]) {
            memset(Next[sz], 0, sizeof(Next[sz]));
            val[sz] = 0;
            flag[sz] = 0;
            id[sz] = 0;
            Next[u][k] = sz++;
        }
        u = Next[u][k];
        val[u]++;
    }
    id[u] = s;
}

void modify(int s, int v) {
    int u = 0;
    for (int i = 31; i >= 0; i--) {
        int k = (s >> i) & 1;
        flag[Next[u][k]] += v;
        u = Next[u][k];
    }
}

int query(int s) {
    int u = 0;
    for (int i = 31; i >= 0; i--) {
        int k = ((s >> i) & 1) ^ 1;
        if (!Next[u][k] || flag[Next[u][k]] >= 1 && val[Next[u][k]] <= 1 || flag[Next[u][k]] == 2 && val[Next[u][k]] <= 2)
            k ^= 1;
        u = Next[u][k];
    }
    return id[u];
}

int main() {
    int t;
    scanf("%d", &t);
    while (t--) {
        init();
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            insert(a[i]);
        }
        int res = 0;
        for (int i = 1; i <= n; i++) {
            modify(a[i], 1);
            for (int j = i + 1; j <= n; j++) {
                int Sum = a[i] + a[j];
                modify(a[j], 1);
                int temp = query(Sum);
                res = max(res, Sum ^ temp);
                modify(a[j], -1);
            }
            modify(a[i], -1);
        }
        printf("%d\n", res);
    }
    return 0;
}