LeetCode每日一题【形成两个异或相等数组的三元组数目】

题目来源

来源：力扣（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;
}