形成两个异或相等数组的三元组数目（数学、前缀和）

形成两个异或相等数组的三元组数目 | 数学 | 数组前缀和

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

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

题目分析：

引入异或数组常用的一个概念：异或数组的前缀和

不妨设数组s[i] = arr[0] ^ arr[1] ^… arr[i-1] ;

a = arr[i] ^ arr[i + 1] ^… arr[j - 1] = s[ j ] ^ s[ i ]

b = arr[j] ^ arr[j + 1] ^… arr[k] = s[k + 1] ^ s[ j ]

因此 a = b 等价于 s[ j ] ^ s[ i ] = s[k + 1] ^ s[ j ] 等价于 s[ i ] = s[k + 1]

---

遍历穷举：

class Solution {
    public int countTriplets(int[] arr) {
        int n = arr.length;
        int[] s = new int[n + 1];
        for(int i = 0;i < n;i++) {
            s[i + 1] = s[i] ^ arr[i];
        }
        int ans = 0;
        for(int i = 0;i < n;i++) {
            for(int j = i + 1;j < n;j++) {
                for(int k = j;k < n;k++) {
                    if(s[i] == s[k + 1]) {
                        ans++;
                    }
                }
            }
        }
        return ans;
    }
}

---

减少一层循环的遍历穷举

经过分析可以看出 a = b 等价于 s[ i ] = s[k + 1]，因此只要 i < j < k +1 均满足题意。

因此，只需要计算 k - i 的值并且累加即可。

class Solution {
    public int countTriplets(int[] arr) {
        int n = arr.length;
        int[] s = new int[n + 1];
        for(int i = 0;i < n;i++) {
            s[i + 1] = s[i] ^ arr[i];
        }
        int ans = 0;
        for(int i = 0;i < n;i++) {
            for(int k = i + 1;k < n;k++) {
                if(s[i] == s[k + 1]) {
                    ans += k - i;
                }
            }
        }
        return ans;
    }
}