LeetCode 每日一题 1442.形成两个异或相等数组的三元组数目

---

问题描述

简要思路

暴力解的话就是三层 for 循环。
这里使用前缀和可简化到 O(n²).
先计算前缀和数组 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;
    }
};