zzuli 1893 985的数学难题

Description

985有n个正整数，他想快速知道下面函数的返回值

int a[N+1];

long long Solve() {

    int i, j;

    long long ans = 0;

    for(i = 1; i <= N; i++) {

     for(int j = i + 1; j <= N; j++) {

    ans += a[i] + a[j] + (a[i] ^ a[j]) + (a[i] | a[j]) + (a[i] & a[j]);

}

    }

    return ans;

}

注：^表示异或运算。

Input

第一行输入一个整数t，代表有t组测试数据。

每组数据第一行输入一个整数代表元素个数，接下来一行输入n个正整数a[]。

注：1 <= t <= 30，1 <= n，a[] <= 100000。

Output

一个整数代表最后的返回值ans。

Sample Input

211021 1

Sample Output

04

这类位运算之前我都没有怎么接触，所以一脸懵逼。

看了网上的才知道 a|b=a^b+a&b;

那么等式就变成 ai+aj+2*a|b;

显然ai+aj.整合一下是 （n-1）a1+a2+....an;

最后只用求a|b就好了。

把每个数的二进制形式分析一下，找出所有有效的1.

（(ans[0][i]) * ans[1][i]) >> 1) * li;      //与运算 

((ans[0][i]) * ans[1][i] + (((ans[1][i]-1) * ant[1][i]) >> 1)) * li;        //或运算   

((ans[1][i]-1) * ant[1][i]) >> 1) * li;       //异或运算  

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;

#define LL long long
#define N 200005
#define mod 1000000007
LL a[N],ans[2][50];
int main()
{
   int t,n,nu;
   scanf("%d",&t);
   while(t--)
   {
       LL sum=0,li=0;
       memset(ans,0,sizeof(ans));
       scanf("%d",&n);
       for(int i=0;i<n;i++)
       {
           scanf("%d",&a[i]);
           sum+=a[i];
       }
       sum*=(n-1);//printf("%lld! %lld\n",sum,li);
       sort(a,a+n);
       for(li;a[n-1]>>li!=0;li++)
       {
           for(int j=0;j<n;j++)
           {
               ans[a[j]>>li&1][li]++;
           }
       }
       for(int i=0;i<li;i++)
       {
           sum+=(1<<i)*(ans[0][i]*ans[1][i]+((ans[1][i]-1)*ans[1][i])/2)*2;
       }
       printf("%lld\n",sum);
   }
   return 0;
}