985的数学难题

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

很显然不能直接写函数计算

这种类型的题目涉及到二进制运算

有定理x^y+x&y==x|y

考虑二进制的每一位对结果的影响

比如1101、1001、1010进行上述运算

则可发现:

从右到左每一位上有1的个数记为

cnt[0]~cnt[3]分别为2、1、1、3

因为或运算除了0|0==0

其余都为1也就是0|1+1|1.

依次看二进制末尾有多少数贡献了1，再根据位运算的性质进行计数即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define size 100009
using namespace std;
int a[size];
int cmp(int a,int b)
{
	return a>b;
}
int main()
{
	int t,i,n;
	scanf("%d",&t);
	while(t--)
	{
		long long sum=0;
		scanf("%d",&n);
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sum+=a[i];
		}
		sum*=(n-1);
		sort(a+1,a+n+1,cmp);
		long long mud=1,ans;
		while(a[1])
		{
		    ans=0;
			for(i=1;i<=n;i++)
			{
				if(a[i]==0)
				break;
				if(a[i]&1)
				ans++;
				a[i]>>=1;//右移一位，二进制最右边的去掉一位。
			}
			sum+=(ans*(n-ans)+(ans-1)*ans/2)*mud*2;
			mud<<=1;//左移一位，变成原来的二倍。
		}
		printf("%lld\n",sum);
	}
	return 0;
}