[codeforces 1368D] AND, OR and square sum 操作过后各个位上1的数量竟然不变

Codeforces Global Round 8  参与排名人数13404

[codeforces 1368D]    AND, OR and square sum   操作过后各个位上1的数量竟然不变

总目录详见https://blog.csdn.net/mrcrack/article/details/103564004

在线测评地址http://codeforces.com/contest/1368/problem/D

Problem	Lang	Verdict	Time	Memory
D - AND, OR and square sum	GNU C++17	Accepted	62 ms	1800 KB

题目大意：给定数组，任选两个元素ai,aj，变更为ai&aj,ai|aj,以上操作可以做任意次。经过变更后，计算ai^2的和，要求和值最大。

样例模拟如下

3
1 3 5

51


1(001),3(011),5(101)

1&3,1|3操作如下
(001)&(011)=(001),(001)|(011)=(011)
操作前1,3
210(表示位置)
001
011
第0位上1的数量是2
第1位上1的数量是1

操作后1&3,1|3
210(表示位置)
001
011
第0位上1的数量是2
第1位上1的数量是1

可以看到：操作过后各个位上1的数量竟然不变

1&5,1|5操作如下
(001)&(101)=(001),(001)|(101)=(101)
操作前1,5
210(表示位置)
001
101
第0位上1的数量是2
第2位上1的数量是1

操作后1&5,1|5
210(表示位置)
001
101
第0位上1的数量是2
第2位上1的数量是1

可以看到：操作过后各个位上1的数量竟然不变

3&5,3|5操作如下
(011)&(101)=(001),(011)|(101)=(111)
操作前3,5
210(表示位置)
011
101
第0位上1的数量是2
第1位上1的数量是1
第2位上1的数量是1

操作后3&5,3|5
210(表示位置)
001
111
第0位上1的数量是2
第1位上1的数量是1
第2位上1的数量是1

可以看到：操作过后各个位上1的数量竟然不变

可以考虑，集中优势兵力，将每位上的1先分配到B[1],剩下的分配到B[2],继续剩下的分配到B[3],以此类推。

AC代码如下

#include <stdio.h>
#define maxn 200010
#define LL long long
int a[maxn],b[maxn],cnt[30];
int main(){
	int i,j,n;
	LL ans=0;
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
		for(j=0;j<20;j++)
			if(a[i]&1<<j)cnt[j]++;//cnt[j]统计j位上1的数量
	for(j=0;j<20;j++)
		for(i=1;i<=cnt[j];i++)
			b[i]+=1<<j;//分配顺序b[1],b[2],b[3],......
	for(i=1;i<=n;i++)
		ans+=(LL)b[i]*b[i];
	printf("%lld\n",ans);
	return 0;
}