LeetCode 1442 形成两个异或相等的三元组数目 题解

LeetCode 1442 形成两个异或相等的三元组数目 题解

给你一个整数数组 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) 的数目。
示例 1：
输入：arr = [2,3,1,6,7]
输出：4
解释：满足题意的三元组分别是 (0,1,2), (0,2,2), (2,3,4) 以及 (2,4,4)

方法：异或前缀和

三重循环
复杂度分析
时间复杂度：O(n^3)，其中 n 是数组 arr 的长度。
空间复杂度：O(n)。

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;
    }
}

二重循环
复杂度分析
时间复杂度：O(n2)，其中 n 是数组 arr 的长度。
空间复杂度：O(n)。

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;
    }
}

一重循环

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];
        }
        Map<Integer, Integer> cnt = new HashMap<Integer, Integer>();
        Map<Integer, Integer> total = new HashMap<Integer, Integer>();
        int ans = 0;
        for (int k = 0; k < n; ++k) {
            if (cnt.containsKey(s[k + 1])) {
                ans += cnt.get(s[k + 1]) * k - total.get(s[k + 1]);
            }
            cnt.put(s[k], cnt.getOrDefault(s[k], 0) + 1);
            total.put(s[k], total.getOrDefault(s[k], 0) + k);
        }
        return ans;
    }
}

class Solution {
    public int countTriplets(int[] arr) {
        int n = arr.length;
        Map<Integer, Integer> count = new HashMap<>();//key为s(i)，value为与s(i)出现的次数
        Map<Integer, Integer> total = new HashMap<>();//key为s(i)，value为与s(i)的值相同的所有i的值的和
        int res = 0, sk = 0;
        for(int k = 0;  k < n; i++){
            s_k1 = sk ^ arr[k];//s(k+1)
            //当k=n-1时， 而i必须小于k,即需要在s(0)~s(n-2)里面找，正好上一个循环里面put进去的是s(n-2)的相关值
            if(count.containsKey(s_k1)){
                res += count.get(s_k1) * k;
                res -= total.get(s_k1)
            }
            count.put(sk, count.getOrDefault(sk, 0) + 1);
            total.put(sk, total.getOrDefault(sk, 0) + k);
            sk = s_k1;
        }
        return res;
    }
}