Codeforces Round #672 (Div. 2) B. Rock and Lever

题目链接

【题意】

给你一个正整数n和一个正整数数组a。任务是计算 ai&aj ≥ ai^a j的对（i，j）的数目。

【思路】

根据题意可以知道，只需要比较每两个数的二进制位数即可。如果两个数的最高位数相同，那么一定满足条件，如果不同，则一定不满足。
有两种做法。

第一种做法：先排序，因为是判断的二进制位，所以只需要每两位进行比较即可。

sort(a+1,a+1+n);
for(ll i=2;i<=n;i++){
    if((a[i]&a[i-1])>=(a[i]^a[i-1])){
        num++;
        ans+=num;
    }
    else num=0;
}

cout << ans <<endl;

这里ans++是因为如果这两位满足条件，如果下一位和该位依旧满足条件，即a[i-1]和a[i]满足条件且a[i]和a[i+1]，那么证明a[i-1]和a[i+1]的二进制位数也是相同的，也满足条件。

第二种做法：记录每个数字的二进制位数，并储存每个位数的数量，最后进行计算。

for (int i = 0; i<n; i++) {
    cin>>a;
    int temp=0;
    while(a) 
        temp++,a/=2;
        b[temp]++;
    }
ll ans=0;
for (int i = 1; i < 50; ++i) 
    ans+=b[i]*(b[i]-1)/2;
    
cout << ans <<endl;