LeetCode-Python-1442. 形成两个异或相等数组的三元组数目 (数组 + 前缀和）

给你一个整数数组 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) 的数目。

 

示例 1：

输入：arr = [2,3,1,6,7]
输出：4
解释：满足题意的三元组分别是 (0,1,2), (0,2,2), (2,3,4) 以及 (2,4,4)
示例 2：

输入：arr = [1,1,1,1,1]
输出：10
示例 3：

输入：arr = [2,3]
输出：0
示例 4：

输入：arr = [1,3,5,7,9]
输出：3
示例 5：

输入：arr = [7,11,12,9,5,2,7,17,22]
输出：8
 

提示：

1 <= arr.length <= 300
1 <= arr[i] <= 10^8

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

第一种思路：

异或的题首先就要回忆异或性质：

1. A ^ A = 0

2. A ^ 0 = A

然后比较直接的思路就是三重循环取两个区间，计算每个区间里的异或结果，按照题意进行比较。

可以稍微优化一点，先得到异或的前缀和数组 leftXOR，leftXOR[ i ] 就是代表从下标为0到 i - 1所有元素的异或之和，

再利用上面的异或性质可以计算区间 [i, j] 的异或之和为 leftXOR[i] ^ leftXOR[j + 1]。

时间复杂度：O(N^3)

空间复杂度：O(N)

class Solution(object):
    def countTriplets(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        leftXOR = [0 for _ in arr] + [0]

        res = 0
        
        for i, x in enumerate(arr):
            leftXOR[i + 1] = leftXOR[i] ^ x

        for i in range(len(arr) + 1):
            for j in range(i + 1, len(arr) + 1):
                for k in range(j, len(arr)):
                    if leftXOR[i] ^ leftXOR[j] == leftXOR[j] ^ leftXOR[k + 1]:
                        res += 1

        return res

第二种思路：

还是异或的性质，题目中的 A == B 这个条件实际上等价于 A ^ B = 0，

所以我们只要找到所有异或之和为 0 的 的子区间，判断中间可以切在几个点上，再累加到答案里即可。

时间复杂度：O（N^2)

空间复杂度：O（1）

class Solution(object):
    def countTriplets(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        res = 0
        for i in range(len(arr)):
            preSum = arr[i]
            for j in range(i + 1, len(arr)):
                preSum ^= arr[j]
                if not preSum:
                    res += j - i

        return res