Codeforces1420 B. Rock and Lever（按二进制位考虑）

题意：

解法：

对于两个数x和y.
考虑最高位:
1.如果x为1且y为1,那么最高位x&y=1,x^y=0,直接满足条件.
2.如果x为0且y为1,或x为1且y为0,那么x&y=0,x*y=1,不满足条件.
3.如果x为0且y为0,那么x&y=0,x*y=0,需要继续判断下面的位.

综上,不难看出满足条件的数对(x,y)一定都是最高位1位置相同.
令cnt[i]表示最高位的1在第i位的数的数量,
那么ans=sum{C(cnt[i],2) }.

code：

#include <bits/stdc++.h>
#define int long long
#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
int a[maxm];
int n;
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    int cnt[33]={0};
    for(int i=1;i<=n;i++){
        for(int j=30;j>=0;j--){
            if(a[i]>>j&1){
                cnt[j]++;
                break;
            }
        }
    }
    int ans=0;
    for(int i=0;i<=30;i++){
        ans+=cnt[i]*(cnt[i]-1)/2;
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    int T;cin>>T;while(T--)
    solve();
    return 0;
}

---