B. Rock and Lever

最新推荐文章于 2021-10-21 22:14:18 发布

原创最新推荐文章于 2021-10-21 22:14:18 发布 · 262 阅读

CC 4.0 BY-SA版权

本文介绍了一种利用位运算解决特定统计问题的方法。通过计算数组中所有元素的最高位，并利用这一信息统计满足特定条件的元素对数量。文章提供了一个完整的C++实现示例。

添加链接描述
题意：给你一个数组，求出 i < j 时， a i & a j ≥ a i ⊕ a j 成立情况的总数。
思路：当时做的时候有思路，就是比较最高位，最高位都是1，则统计，但是不知道位运算怎么写，就在此记录一下。
算最高位是第几位的代码

for(int i=0;i<n;i++)
        {
            ll st=1,pos=0;
            while(st<=a[i])
            {
                st*=2;
                pos++;
            }
            ans+=cur[pos];
            cur[pos]++;
        }

其实就是算2的幂，大于它的时候停止。
总的代码：

#include<bits/stdc++.h>
#define ll long long
#define ios ios::sync_with_stdio(false);cin.tie(0); cout.tie(0);
using namespace std;
const int maxn=1e5+5,INF=0x3f3f3f3f;
ll a[maxn],cur[maxn];
int main()
{
    ios;
    int t;
    cin>>t;
    while(t--)
    {
        memset(cur,0,sizeof(cur));
        int n;
        ll ans=0;
        cin>>n;
        for(int i=0;i<n;i++)
            cin>>a[i];
        for(int i=0;i<n;i++)
        {
            ll st=1,pos=0;
            while(st<=a[i])
            {
                st*=2;
                pos++;
            }
            ans+=cur[pos];
            cur[pos]++;
        }
        cout<<ans<<endl;
    }
    return 0;
}