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 a,b a,b同时异或 b b b就会发现,只要 i i i到 k k k的元素异或等于0,那么 j j j取 i , k i,k 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