问题描述
简要思路
暴力解的话就是三层 for 循环。
这里使用前缀和可简化到 O(n2).
先计算前缀和数组 s[ ]
然后考虑查找条件:这里需要寻找的区 [ i , j - 1 ] 和 [ j , k ] 的异或结果相同
即 (s[ j - 1 ] ^ s [ i - 1 ]) ^ (s[ k ] ^ s [ j - 1 ]) == 0,简化可得
s [ i - 1 ] = s[ k ]
故双循环寻找 s[ i - 1 ] == s[ k ] 即可。且在 [ i - 1, k ] 区间内的所有 j 都满足。所以 ans += k - i
0ms 代码。
代码
class Solution {
public:
int countTriplets(vector<int>& arr) {
int n = arr.size();
if(n < 2) return 0;
vector<int> s(n+1, 0);
for(int i = 1; i <= n; i++){
s[i] = arr[i-1] ^ s[i-1];
}
int ans = 0;
for(int i = 1; i < n; i++){
for(int k = i+1; k <= n; k++){
if(s[i-1] == s[k]) ans += k - i;
}
}
return ans;
}
};