题目来源
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor
题目描述
给你一个整数数组 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) 的数目。
c语言题解
int countTriplets(int* arr, int arrSize){
int a[arrSize+1];
int i, j, k;
a[0] = 0;
int count;
/*
arr = [a,b,c,d]
a[0] = 0
a[1] = a
a[2] = a^b
a[3] = a^b^c
a[4] = a^b^c^d
*/
for(i = 0; i<arrSize; i++){
a[i+1] = a[i]^arr[i];
}
for(i = 0; i<arrSize-1; i++){
for(j = i+1; j<arrSize; j++){
for(k = j; k<arrSize; k++){
/*i = 1, j = 2, k = 3
a[i] = a; a[j] = a^b; a[k] = a^b^c; a[k] = a^b^c^d
A = b, B = c^d
a == a^b^c^d --> b^c^d = 0 --> A^B = 0 --> A == B = 0
*/
if(a[i] == a[k+1]) count++;
}
}
}
return count;
}