ZYB loves Xor I(hud5269)

ZYB loves Xor I

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

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

思路:字典树；

题意：

ZYB喜欢研究Xor，现在他得到了一个长度为nn的数组A。于是他想知道：对于所有数对(i,j)(i in [1,n],j in [1,n])(i,j)(i∈[1,n],j∈[1,n])，lowbit(A_i xor A_j)lowbit(A​i​​xorA​j​​)之和为多少.由于答案可能过大，你需要输出答案对998244353取模后的值
定义lowbit(x)=2^k2​k​​，其中k是最小的满足(xx andand 2^k)>02​k​​)>0的数
特别地：lowbit(0)=0
                                              
将数转化为二进制，我们只需要枚举第k位满足亦或为1，那么分别统计这位上是0 x1，和1 x2，并且保证在x1,x2在[k-1,0]这些位上是相同的，那么我们可以建立字典树来统计字典树保证了当前节点前的所有位都相同。复杂度O(n*30);

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<string.h>
#include<math.h>
#include<stdlib.h>
using namespace std;
typedef long long LL;
const LL mod = 998244353;
struct node
{
    node *p[2];
    LL val;
    node()
    {
        memset(p,0,sizeof(p));
        val = 0;
    }
};
int a[100005];
short int id[100005][30];
int si[100005];
node *head;
void fr(node *h);
void in(node *h,int num);
LL ask(node *h,LL x);
int main(void)
{
    int n,m,t,__cn = 0;;
    scanf("%d",&t);
    while(t--)
    {
        memset(id,0,sizeof(id));
        scanf("%d",&n);
        head = new node();
        for(int i = 1; i <= n; i++)
        {
            scanf("%d",&a[i]);
            int t = 0;
            int tt = 29;
            while(tt)
            {
                id[i][t++] = a[i]%2;
                a[i]/=2;
                tt--;
            }
            si[i] = t;
            in(head,i);
        }
        LL sum = ask(head,1);
        fr(head);
        printf("Case #%d: ",++__cn);
        printf("%lld
",sum);
    }
    return 0;
}
void in(node *h,int  num)
{
    int i,j;
    node *ak = h;
    for(i = 0; i < si[num]; i++)
    {
        int c = id[num][i];
        if(ak->p[c]==NULL)
        {
            ak->p[c] = new node();
        }
        ak->p[c]->val++;
        //printf("%d",c);
        ak = ak->p[c];
    }
    //printf("
");
}
void fr(node *h)
{
    int i;
    for(i = 0; i < 2; i++)
    {
        if(h->p[i]!=NULL)
        {
            fr(h->p[i]);
        }
    }
    free(h);
}
LL ask(node *h,LL x)
{
    node *ak = h;
    int i,j;
    LL sum = 0;
    if(h->p[0]!=NULL&&h->p[1]!=NULL)
    {
        LL ac = (1<<x)%mod;
        sum = sum + (ac*(h->p[0]->val*h->p[1]->val)%mod)%mod;
        sum%=mod;
    }
    for(i = 0; i < 2; i++)
    {
        if(h->p[i]!=NULL)
        {
            sum = (sum+ask(h->p[i],x+1))%mod;
        }
    }
    return sum;
}