- 形成两个异或相等数组的三元组数目
给你一个整数数组 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/