给你一个数组A [1..n]
,你需要计算多少三元组(i,j,k)
满足(i <j <k)
和((A [i] xor A [j])<(A [j] xor A [k]))
样例
样例 1:
输入:[1,2,3,4,5]
输出:6
解释:[1,2,4],[1,2,5],[1,3,4],[1,3,5],[2,3,4],[2,3,5]符合要求。
样例 2:
输入:[1,2,3]
输出:0
解释:[1,2,3]不符合要求。
注意事项
1≤n≤5∗10^5
0≤A[i]<2^30
输入测试数据 (每行一个参数)如何理解测试数据?
32位,计算2对xor操作得到的数,从哪一位开始不同的。对于计算每一位时,计算每一位上的每一个数时,只需计算它之前 和 之后 最后一位不同的对数(0——>1,1——>0)n0,n1。
public class Solution {
/**
* @param a: the array
* @return: how many tuples in the array
*/
public long getCount(int[] a) {
long cnt=0;
int len=a.length;
for(int k=31;k>=0;k--){//计算从哪一位开始不同的
HashMap<Integer,Long> pre=new HashMap<>();
HashMap<Integer,Long> post=new HashMap<>();
long n0=0,n1=0;//0-1对,1-0对
for(int i=1;i<len;i++){
int c=a[i]>>k;
post.put(c,post.getOrDefault(c,0L)+1);
}
for(int i=1;i<len-1;i++){
int c1=a[i-1]>>k;
int c2=a[i]>>k;
if((c2&1)==0){//0,0,1
n1-=pre.getOrDefault(c2+1,0L);
}else{
n0-=pre.getOrDefault(c2-1,0L);
}
pre.put(c1,pre.getOrDefault(c1,0L)+1);
post.put(c2,post.getOrDefault(c2,0L)-1);
if((c1&1)==0){//0-1对
n0+=post.getOrDefault(c1+1,0L);
}else{
n1+=post.getOrDefault(c1-1,0L);
}
if((c2&1)==0){
cnt+=n0;
}else{
cnt+=n1;
}
}
}
return cnt;
}
}