小G的LY数对 (手写hash表）

题目链接

题意：
小G定义LY数对为两个数x，y在二进制的异或操作后恰好有两位是1
小G现在有两个数组a,b长度分别为n,m
现在小G想知道有多少对i,j满足
(1<=i<=n,1<=j<=m)满足a[i]和b[j]是LY数对

思路：既然需要有两位不一样，那么改变a的一位如果与改变了一位的b相同，说明它们为其中一对，但是如果a与b直接相等我们不能计入答案，所以需要删除出来。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int mod=1000007;
int a[300010],b[300010];
struct tt{
    int x,to,cnt;
};
tt e[3000010];
int h[3000010];
int idx=1;
int n,m;
void insert(ll x){
    int k=(x%mod+mod)%mod;
    for(int i=h[k];i!=-1;i=e[i].to){
        if(e[i].x==x){
            e[i].cnt++;
            return;
        }
    }
    e[idx].x=x;
    e[idx].cnt=1;
    e[idx].to=h[k];
    h[k]=idx++;
}
ll find(ll x){
    int k=(x%mod+mod)%mod;
    for(int i=h[k];i!=-1;i=e[i].to){
        if(e[i].x==x){
            return e[i].cnt;
        }
    }
    return 0;
}
int main(){
    memset(h,-1,sizeof h);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d",&b[i]);
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<30;j++){
            insert(a[i]^(1<<j));
        }
    }
    ll ans=0;
    for(int i=1;i<=m;i++){
        for(int j=0;j<30;j++){
            ans+=1ll*find(b[i]^(1<<j));
        }
    }
    memset(e,0,sizeof e);
    memset(h,-1,sizeof h);
    idx=1;//再次使用需要情况idx
    for(int i=1;i<=n;i++){
        insert(a[i]);
    }
    for(int i=1;i<=n;i++){
        ans-=1ll*find(b[i])*30;//删除两个数相同的贡献，因为每个位变化都会产生1，所以需要乘以30
    }
    printf("%lld\n",ans>>1);
}