AtCoder 3943 Two Sequences

題面

題意

給出兩個序列a，b，每個序列中都有n個數，將它們兩兩相加，一共得到n^2個和，求這幾個和的異或值。

做法

考慮逐位確定答案，首先可以發現a,b兩數相加後，如果表示m的這位是1，那麼令c=a%(m<<1),d=b%(m<<1),則m<=a+b<(m<<1)或a+b>=(m*3).
然後逐位考慮，將b序列模（m*2）後排序，那麼在b序列中與a[i]之和表示m的這位是1就是2段區間，二分即可。

代碼

#include<iostream>
#include<cstdio>
#include<algorithm>
#define N 200100
using namespace std;

int n,a[N],b[N],M,l,r,mid,ans,an;

inline int find(int u)
{
    int l=1,r=n+1,mid;
    for(;l<r;)
    {
        mid=((l+r)>>1);
        if(b[mid]>=u) r=mid;
        else l=mid+1;
    }
    return l;
}

int main()
{
    int i,j,p,q;
    cin>>n;
    for(i=1;i<=n;i++) scanf("%d",&a[i]);
    for(i=1;i<=n;i++) scanf("%d",&b[i]);
    for(M=(1 << 29),i=(1 << 28);M>=2;M>>=1,i>>=1)
    {
        an=0;
        for(j=1;j<=n;j++)
        {
            a[j]%=M;
            b[j]%=M;
        }
        sort(b+1,b+n+1);
        for(j=1;j<=n;j++)
        {
            an^=(((find(2*i-a[j])-find(i-a[j]))+(n-find(3*i-a[j])+1))&1);
        }
        an&=1;
        ans+=an*(M>>1);
    }
    cout<<ans;
}