hdu 5269 trie树

现场想到了lowbit(X xor Y)=X和Y从右向左数，对应相同的数的个数+1。。。然而并没有想到接下来用trie树

然后就想排个序试试。。。然后就整个人都不好了啊摔

 

sol：用trie，一边insert一边输出答案

找出Ai的二进制数翻转后在字母树上的路径，对于路径上每个点x，设他走的边是v，且当前为第k位，则和他xor后lowbit为2k的数的个数为trans(x,v^1)的子树大小。

注意这里v=0or1，即可以取的字符。

仿照lrj的模板写了一次。。结果写残了233。因为没有考虑好0这种情况

lrj模板里插入的都是字符串，不会存在0（也就是空）的情况。然而要想插入数字0就完蛋了

    int Insert(LL X)
    {
        int u=0,dep=1;
        while(X!=0)
        {
            int c=(X&1)+1;
            X=X>>1;
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                ch[u][c]=sz;
                sz++;
            }
            val[u]++;
            dp[u]=dep;
            dep++;
            f[ch[u][c]]=u;
            u=ch[u][c];
        }
        return u;
    }

 

 

 

AC Code：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define LL long long
#define MOD 998244353;
LL bit[30];

int a[100010];
LL ans ;
int T;

struct Trie{
    int next[30*50010][2];
    int num[30*50010];
    int root,tot;
    int newnode(){
        num[tot] = 0;
        next[tot][0] = next[tot][1] = -1;
        tot++;
        return tot-1;
    }
    void init(){
        tot = 0;
        root = newnode();
    }
    void insert(int val){
        int now = root;
        for(int i = 0;i < 30;i++){
            int id;
            if(val&bit[i])id = 1;
            else id = 0;
            if(next[now][id^1] != -1){
                //ans += (num[next[now][id^1]]<<i)%MOD;
                ans += (long long)bit[i]*num[next[now][id^1]]%MOD;
                ans %= MOD;
            }
            if(next[now][id] == -1)
                next[now][id] = newnode();
            now = next[now][id];
            num[now]++;
        }
    }
}tree;

int main()
{
    bit[0] = 1;
    for(int i = 1;i < 30;i++)
        bit[i] = 2*bit[i-1];

    cin>>T;
    for(int TT=1;TT<=T;TT++)
    {
        int n;
        cin>>n;
        ans = 0;
        tree.init();
        for(int i = 1;i <= n;i++)
        {
            scanf("%d",&a[i]);
            tree.insert(a[i]);
        }
        ans = ans*2%MOD;
        printf("Case #%d: %I64d\n",TT,ans);
    }
    return 0;
}