LeetCode 1442 形成两个异或相等数组的三元组数目[异或 位运算 数学] HERODING的LeetCode之路

Posted 2021-05-20 HERODING23

解题思路：
既然a和b相等，且a的异或序列和b的异或序列紧紧相连，那么我们就有理由知道a到b结束的序列的异或结果是0，那么直接子序列的异或值为0的序列即可，那么数量如何统计呢？观察可以发现一串异或值为0的序列，无论分割位置k在哪里，所划分的a和b的异或值都相等，这样结果直接加 (j - i) 即可，代码如下：

class Solution {
public:
    int countTriplets(vector<int>& arr) {
        // 定义长度，数量
        int len = arr.size();
        int ans = 0;
        for(int i = 0; i < len - 1; i ++) {
            // 异或结果
            int sum = 0;
            for(int j = i; j < len; j ++) {
                sum ^= arr[j];
                // 结果为 0 且 j 比 i 大（说明不是原地不动）
                if(sum == 0 && j > i) {
                    ans += (j - i);
                }
            }
        }
        return ans;
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/count-triplets-that-can-form-two-arrays-of-equal-xor/solution/cqiao-jie-by-heroding-we23/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/