a = b 等价于 a ^ b = 0。
那么我们的问题就转换为了arr[i] ^ arr[i+1] ^ … ^arr[k] == 0,可以看到与j的取值无关,只要 j ∈ ( i , k ] j\in(i,k] j∈(i,k]即可满足条件。那么我们就将循环从三层循环降为两层循环,i和k,使用x计算arr[i] ^ arr[i+1] ^ … ^arr[k] 。如果 sum = 0,那么这种情况下就有 k-i个满足条件的(i,j,k)。
class Solution {
public int countTriplets(int[] arr) {
int res=0;
for(int i=0;i<arr.length;i++){
int sum=arr[i];
for(int k=i+1;k<arr.length;k++){
sum^=arr[k];
if(sum==0)
res+=k-i;
}
}
return res;
}
}
另外一种解法则是前缀异或和
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;
}
}