Leetcode 1442.形成两个异或相等数组的三元组数目

1 题目描述（Leetcode题目链接）

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

2 题解

  我们将$a,b$同时异或$b$就会发现，只要$i$到$k$的元素异或等于0，那么$j$取$i,k$之间的任意值都可以满足条件。

class Solution: def countTriplets(self, arr: List[int]) -> int: res = 0 for i in range(len(arr)-1): s = arr[i] for k in range(i+1, len(arr)): s ^= arr[k] if s == 0: res += k - i return res