题目传送门
题意: 给你n个数,对于i!=j,a[i]=x,a[j]=y,我们可以进行一种操作:a[i]=x&y,a[j]=x|y,你可以无限的进行这个操作,问你最后Σa[i] (1<=i<=n) 的最大值是多少。
思路: 我们每次贪心,每次拿出一个当前能拿出的最大的数,那肯定是用其他数和一个数取或运算,然后其他数就被他取与运算了,对于一个位上,有四种情况,(1|1 1&1) ,(1|0 ,1&0) ,(0|1,0&1), (0|0,0&0),我们发现在这个运算时,1的个数是不会变的所以我们要用每个位上有限的1的数量,去组合成n个数使得平方和最大。
贪心策略:前面我们知道了运算之后1的个数不会变,然后我们每次把还有剩余1的位都拿一个给一个数,这样从局部最优到整体最优。
代码:
#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair<int,int>
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=2e5+5;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const double eps=1e-7;
const double PI=acos(-1);
int cnt[50],a[N];
signed main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
int p=0;
while(x)
{
cnt[p]+=x%2;
x>>=1;
p++;
}
}
int res=0;
for(int i=1;i<=n;i++)
{
for(int j=0;j<=30;j++)
{
if(cnt[j])
{
a[i]+=(1<<j);
cnt[j]--;
}
}
res+=a[i]*a[i];
}
cout<<res<<endl;
}