JZOJ4446. 【HNOI2016模拟4.14】B

分析

通过观察样例，推理得出结论：改变之后的平方和更大。
认真分析改变两个数的意义：
就是将小的数二进制某些位的1移到大的那个数那里，然后这个位置变为0。
我们也知道，一个数越大，它的平方越大。

我们只需要记录二进制下每一个位置一共有多少个1，然后每次取最大的数，
这样的答案一定是最优的。

code

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#define N 100003
#define ll long long 
using namespace std;
ll a[N][23],z[23],sum,ans;
int num[23],n,x,now;
char ch;
void read(int &n)
{
    n=0;
    ch=getchar();
    while((ch<'0' || ch>'9') && ch!='-')ch=getchar();
    int w=1;
    if(ch=='-')w=-1,ch=getchar();
    while('0'<=ch && ch<='9')n=n*10+ch-'0',ch=getchar();
    n*=w;
}
int main()
{
    z[0]=1;
    for(int i=1;i<20;i++)
        z[i]=z[i-1]*2;
    read(n);
    for(int i=1;i<=n;i++)
    {
        read(x);
        for(int j=0;j<20;j++)
            if((x&z[j])>0)num[j]++;
    }
    for(int k=19;k>=0;k--)
        for(int i=1;i<=num[k];i++)
            a[i][k]=1;
    for(int i=1;i<=n;i++)
    {
        sum=0;
        for(int j=0;j<20;j++)
            sum+=z[j]*a[i][j];
        ans+=sum*sum;
    }
    printf("%lld\n",ans);
}