dtoj2623. B

给定 n 个数,每次你可以选择两个数 a,b ,将其中一个数变为 a and b ,另一个变成 a or b ,请最大化所有数的平方和。


Sol

由于操作之后a+b是定值,我们希望最大化a^2+b^2,也就希望最大化两个数的差。

把a排序,

按位考虑,每次把最大有无2^i的和最小的有2^i交换i位。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,a[100005],Max;
bool cmp(int x,int y){return x>y;}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    sort(a+1,a+n+1,cmp);
    int W=0;for(;(1<<W)<=a[1];W++);
    
    for(;W>=0;W--){
        int l=1,r=n;
        while(l<r){
            while(l<r&&(a[l]&(1<<W)))l++;
            while(l<r&&!(a[r]&(1<<W)))r--;
            if(l==r)break;
            a[l]+=(1<<W);a[r]-=(1<<W);
            
        }
    }
    long long ans=0;
    for(int i=1;i<=n;i++)ans=ans+1LL*a[i]*a[i];
    cout<<ans<<endl;
    return 0;
}
View Code
原文地址:https://www.cnblogs.com/liankewei/p/12368974.html