HDU_4825_Xor Sum(字典树)

题型:数据结构


题意:中文题目,不解释~


分析:

看一下数据量,数据量10^5,查询量10^5,所以每次O(n)的找最值是不可行的,得先对n个数做一个预处理。

如何预处理呢?
由于寻求的是异或值最大,所以根据异或的性质,"相同为0,不同为1",所以在二进制中,越高位不相同值越大。

因为数据大小不会超过2^32,可以将n个数据转换成32位二进制串,然后由高位到低位构建一棵01字典树,这样,在查询的时候,若该位置为0,就找树上该位置的1节点,但是如果没有1节点,就只好走0节点,反之亦然。

总时间复杂度为O(32*n)。


代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#define LL __int64
using namespace std;

struct Node {
    int num;
    Node *next[2];
    Node() {
        num = 0;
        for(int i=0; i<2; i++) {
            next[i] = NULL;
        }
    }
};
char ans[123];
int n,m;

void tstrrev(char s[]) {
    int l=strlen(s);
    for(int i=0,j=l-1; i<l; i++,j--) {
        if(i<j) {
            s[i]^=s[j];
            s[j]^=s[i];
            s[i]^=s[j];
        }
    }
}

void insert(Node *root,char *s) {
    Node *p;
    p = root;
    for(int i=0; s[i]; i++) {
        int x = s[i] - '0';
        if(p->next[x] == NULL)
            p->next[x] = new Node;
        p = p->next[x];
        p->num ++;
    }
}

void search(Node *root,char *s) {
    memset(ans,0,sizeof(ans));
    Node *p;
    p = root;
    for(int i=0; i<32; i++) {
        int x = s[i] - '0';
        if(p->next[x] != NULL) {
            //if(P->next[x]->num != 0)
            p = p->next[x];
            ans[i] = s[i];
        } else {
            if(x == 1) x = 0;
            else x = 1;
            p = p->next[x];
            ans[i] = x + '0';
        }
    }
}

int main() {
    int _;
    scanf("%d",&_);
    for(int cas=1; cas<=_; cas++) {
        scanf("%d%d",&n,&m);
        LL tmp;
        Node *root = new Node;
        while(n--) {
            scanf("%I64d",&tmp);
            char str[40];
            for(int i=0; i<32; i++) {
                str[i] = tmp % 2 + '0';
                tmp /= 2;
            }
            str[32] = '\0';
            tstrrev(str);
            //插入字典树
            insert(root,str);
        }

        printf("Case #%d:\n",cas);
        while(m--) {
            scanf("%I64d",&tmp);
            char str[40];
            for(int i=0; i<32; i++) {
                str[i] = tmp % 2 + '0';
                tmp /= 2;
            }
            str[32] = '\0';
            tstrrev(str);

            for(int i=0; i<32; i++) {
                if(str[i] == '1') {
                    str[i] = '0';
                } else {
                    str[i] = '1';
                }
            }

            search(root,str);

            tstrrev(ans);

            LL hehe = 0;
            LL mul = 1;
            for(int i=0; i<32; i++) {
                hehe += mul * (ans[i] - '0');
                mul *= 2;
            }
            printf("%I64d\n",hehe);
        }

    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值