Leetcode 1442.形成两个异或相等数组的三元组数目（Count Triplets That Can Form Two Arrays of Equal XOR）

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