码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 5269 ZYB loves Xor I(字典树)

时间:2015-06-16 07:59:37      阅读:106      评论:0      收藏:0      [点我收藏+]

标签:

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5269

思路分析:当lowbit(AxorB)=2p 时,表示A与B的二进制表示的0-p-1位相等,第p位不同;考虑维护一棵字母树,将所有数字

转换为二进制形式并且从第0位开始插入树中,并在每个节点中记录通过该结点的数字数目;最后统计答案,对于每一个数字,

对于在其路径中的每一个结点X,假设其为第K层,统计通过与该结点不同的值的结点的数目count,则结果增加count*2k; 

 

代码如下:

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

const int MAX_N = 5 * 10000 + 100;
int num[MAX_N];
struct Node{
    Node *child[2];
    int count;
    Node(){
        count = 0;
        memset(child, 0, sizeof(child));
    }
    Node(int value){
        count = value;
        memset(child, 0, sizeof(child));
    }
};

void Insert(Node *head, int value){
    Node *pre = head, *next = NULL;
    for (int i = 0; i < 30; ++ i){
        if (pre->child[value & 1] == NULL){
            pre->child[value & 1] = new Node(1);
            pre = pre->child[value & 1];
        }else{
            next = pre->child[value & 1];
            next->count++;
            pre = next;
        }
        value >>= 1;
    }
}

void MakeEmpty(Node *node){
    node->count = 0;
    if (node->child[0])
        MakeEmpty(node->child[0]);
    if (node->child[1])
        MakeEmpty(node->child[1]);
}

long long Query(Node *head, int value){
    Node *pre = head, *next = NULL, *other;
    long long ret = 0;

    for (int i = 0; i < 30; ++ i){
        next = pre->child[value & 1];
        other = pre->child[(value & 1) ^ 1];
        if (other)
            ret = (ret + other->count * (1 << i)) % 998244353;
        pre = next;
        value >>= 1;
    }
    return ret;
}

int main(){
    int case_times, n;
    int case_id = 0;

    scanf("%d", &case_times);
    while (case_times--){
        Node *head = new Node();

        scanf("%d", &n);
        for (int i = 0; i < n; ++i){
            scanf("%d", &num[i]);
            Insert(head, num[i]);
        }

        long long ans = 0;
        for (int i = 0; i < n; ++i)
            ans = (ans + Query(head, num[i])) % 998244353;
        MakeEmpty(head);
        printf("Case #%d: %I64d\n", ++case_id, ans);
    }
    return 0;
}

hdu 5269 ZYB loves Xor I(字典树)

标签:

原文地址:http://www.cnblogs.com/tallisHe/p/4579756.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!