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

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) 的数目。

lass Solution {
public:
    int countTriplets(vector<int>& arr) {
        int n = arr.size();
        // 第一个哈希表
        // 其中有初始边界值 (0, 1)，表示异或和为 0（左边界，一个数都不选择）出现了 1 次
        unordered_map<int, int> freq = {{0, 1}};
        // 第二个哈希表
        // 其中有初始边界值 (0, 0)，表示异或和为 0（左边界，一个数都不选择）对应的 i 值为 0
        unordered_map<int, int> idsum = {{0, 0}};
        int xorsum = 0, ans = 0;
        // 枚举 k
        for (int k = 0; k < n; ++k) {
            // 计算前缀异或和
            xorsum ^= arr[k];
            // 如果这个前缀异或和之前出现过，那么就找到了一些满足要求的三元组
            if (freq.count(xorsum)) {
                // 对应了题解中的公式 t * k - sum(i_t)
                ans += freq[xorsum] * k - idsum[xorsum];
            }
            // 更新前缀异或和的出现次数
            ++freq[xorsum];
            // 更新前缀异或和出现位置的下标之和
            // 注意 i-1 和 i 的关系，所以这里要额外增加 1
            idsum[xorsum] += (k + 1);
        }
        return ans;
    }
};

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor/solution/xing-cheng-liang-ge-yi-huo-xiang-deng-sh-jud0/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor/solution/xing-cheng-liang-ge-yi-huo-xiang-deng-sh-jud0/