LeetCode每日一题 1442. 形成两个异或相等数组的三元组数目

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

难度：中等
语言：java

题目内容

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

题目解答

又又又是异或，最近怎么一直是异或题目啊
看一下题目，a == b，由异或的特性可知，a xor b = 0
那么就是从 i 到 k 的异或为0，这个时候，在[i , k]内 任意数的异或的值一定都等于剩下的异或的值，因为异或满足交换律，这么一想就非常简单了。
只要用两次遍历，找到所有满足条件的 i 与 k，j等于多少并不重要，代码如下

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

看一下答案的代码，用的是前缀异或，与前缀和比较类似，前缀法经常用来解决，求数组中间一部分值的题目，之前做过前缀和，中间[ i, j ]部分的和即为，前缀数组的第 j 项减去前缀数组的第 i 项，这边异或也是一样，中间[ i, j ]部分的异或的值即为前缀数组的第 j 项异或前缀数组的第 i 项，可以用遍历解决问题。

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

后面引入了哈希表的做法，简单点来说就是因为必然有两个值是相等的才能使结果为0，就用哈希表来记录异或的值，那么只需要查询哈希表，就可以得到最后的值了，代码如下

class Solution {
    public int countTriplets(int[] arr) {
        int n = arr.length;
        // 预处理前缀异或数组
        int[] sum = new int[n + 1];
        for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] ^ arr[i - 1];
        int ans = 0;
        // 记录出现过的异或结果，存储格式：{ 异或结果 : [下标1, 下标2 ...] }
        Map<Integer, List<Integer>> map = new HashMap<>();
        for (int k = 0; k <= n; k++) {
            List<Integer> list = map.getOrDefault(sum[k], new ArrayList<>());
            for (int idx : list) {
                int i = idx + 1;
                ans += k - i;
            }
            list.add(k);
            map.put(sum[k], list);
        }
        return ans;
    }
}

总结一下就是，后面的两个方法，比较具有普遍性，也是比较好总结的，但是我觉得这样就失去做题的乐趣了，用固定的方法去解决题目，比的是题海战术，还是发现新的方法会比较好玩一点。