力扣1442.形成两个异或相等数组的三元组数目

阳光男孩01

已于 2024-10-12 15:41:56 修改

阅读量236

点赞数 1

文章标签： leetcode 算法职场和发展

于 2024-08-25 11:14:42 首次发布

本文链接：https://blog.csdn.net/Pisasama/article/details/141526962

版权

力扣1442.形成两个异或相等数组的三元组数目

题目解析及思路

题目要求返回满足a[i] - a[j-1] 异或和 = a[j] - a[k]异或和的三元组数目，其中j可以=k

三元组要求可以转化为a[i] - a[k]的异或和为0

因此用前缀和求异或和，枚举满足条件的ik，j在[i+1,k]都成立

同余
- 如果s[i-1] ^ s[k] = 0，则[i,k]区间的异或和为0
- 即s[i-1] == s[k]
- j在哪不重要，只要是[i,k]内即可

代码

class Solution {
public:
    int countTriplets(vector<int>& arr) {
        int n = arr.size();
        vector<int> s(n+1);
        int ans=0; 
        for(int i=0;i<n;i++)
            s[i+1] = s[i] ^ arr[i];
        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;
    }
};