LC1442. 形成两个异或相等数组的三元组数目
给你一个整数数组 arr 。现需要从数组中取三个下标 i、j 和 k ,其中 (0 <= i < j <= k < arr.length) 。
a 和 b 定义如下:
a = arr[i] ^ arr[i + 1] ^ … ^ arr[j - 1]
b = arr[j] ^ arr[j + 1] ^ … ^ arr[k]
注意:^ 表示 按位异或 操作。
请返回能够令 a == b 成立的三元组 (i, j , k) 的数目。来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
两个异或相等的数组a和b的异或结果为0,即a^b=0,那么可以通过两层循环找到异或结果为0的连续子数组c,数组c的长度为l,根据排列组合易知有l-1对划分组合满足a=b,故总数目加上l-1,代码如下:
int countTriplets(vector<int>& arr) {
int n=arr.size();
int ans=0;
for(int i=0;i<n-1;i++){
int sum=0;
for(int k=i;k<n;k++){
sum^=arr[k];
if(k>i&&sum==0){
ans+=k-i;
}
}
}
return ans;
}