HDU - 5269【SBBBBBB Trie】

ZYB loves Xor I

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 979    Accepted Submission(s): 436

Problem Description

Memphis loves xor very musch.Now he gets an array A.The length of A is n.Now he wants to know the sum of all (lowbit(Ai xor Aj)) (i,j∈[1,n])
We define that lowbit(x)=2k,k is the smallest integer satisfied ((x and 2k)>0)
Specially,lowbit(0)=0
Because the ans may be too big.You just need to output ans mod 998244353

 

 

Input

Multiple test cases, the first line contains an integer T(no more than 10), indicating the number of cases. Each test case contains two lines
The first line has an integer n
The second line has n integers A1,A2....An
n∈[1,5∗104]，Ai∈[0,229]

 

 

Output

For each case, the output should occupies exactly one line. The output format is Case #x: ans, here x is the data number begins at 1.

 

 

Sample Input

2 5 4 0 2 7 0 5 2 6 5 4 0

 

 

Sample Output

Case #1: 36 Case #2: 40

 
题意：求∑ lowbit(a xor b) a,b属于这n个数。(a, b)和(b, a)算两次。
题解：观察发现对于任意一个数 x 来说，只要前 i-1 位和其它数相等，第i位不相等。那么 ans += cnt * pow(2, i-1);
           cnt 为和它前i-1个前缀都相等的数的个数。 可以利用字典树边插入边更新ans，最后ans*2就是答案。
          ps:  SB 错误错了一天。。。#define N 50000+5 那个开数组的时候 N*33就变成了 50000+5*33。就错了一天！！！！！
代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <set>
#include <map>
#define rep(i,a,b) for(int i = a;i <= b;++ i)
#define per(i,a,b) for(int i = a;i >= b;-- i)
#define mem(a,b) memset((a),(b),sizeof((a)))
#define FIN freopen("in.txt","r",stdin)
#define FOUT freopen("out.txt","w",stdout)
#define IO ios_base::sync_with_stdio(0),cin.tie(0)
#define mid ((l+r)>>1)
#define ls (id<<1)
#define rs ((id<<1)|1)
#define N 50005
#define INF 0x3f3f3f3f
#define INFF 0x3f3f3f3f3f3f3f
#define mod 998244353
typedef long long ll;
using namespace std;

int T,n;
ll x,ans,qpow[60];
struct Trie{
    int ch[N*33][2], val[N*33], tol;
    void Init()    { mem(ch[0], -1); mem(val, 0); tol = 1; }

    void insert(ll x){
        int u = 0;
        rep(i, 0, 29){
            int v = (x&1);
            if(ch[u][v^1] != -1){
                ans += qpow[i] * val[ch[u][v^1]];
                if(ans > mod)    ans %= mod;
            }
            if(ch[u][v] == -1){
                mem(ch[tol], -1);
                ch[u][v] = tol++;
            }
            x >>= 1;
            u = ch[u][v];
            val[u]++;
        }
    }
}trie;
void fuc(){
    qpow[0] = 1;
    rep(i, 1, 30)    qpow[i] = (qpow[i-1]%mod * 2)%mod;    
}
int main()
{IO;
    fuc();
    //FIN;
    cin >> T;
    int w_w = 0;
    while(T--){
        cin >> n;
        
        ans = 0;
        trie.Init();
        rep(i, 1, n){
            cin >> x;
            trie.insert(x);
        }
        cout << "Case #" << ++w_w << ": " << ans*2%mod << endl;
    }
    return 0;
}