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;
}