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