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

最新推荐文章于 2024-07-14 02:25:25 发布

csu_xiji

最新推荐文章于 2024-07-14 02:25:25 发布

阅读量102

点赞数

分类专栏：力扣位运算哈希

本文链接：https://blog.csdn.net/xiji333/article/details/117000681

版权

力扣同时被 3 个专栏收录

282 篇文章 2 订阅

订阅专栏

哈希

32 篇文章 1 订阅

订阅专栏

位运算

24 篇文章 0 订阅

订阅专栏

在这里插入图片描述
思路：设 $S_0=0,S_i=a_0\bigoplus a_1\bigoplus……\bigoplus a_{i-1}$ ，由异或性质可知，数组 $a r r$ 的 $[i, j - 1]$ 的异或和等于 $S_j \bigoplus S_i$ ，那么由题目的 $a = b$ 可以化简得到 $S_i=S_{k+1}$ ，此时 $j$ 有 $k + 1 - i$ 种选择，它们都是符合题意的解。所以我们可以 $O(n^2)$ 枚举 $i 、 k$ 计算答案。更优的解决方案需要用到哈希表，假设我们在枚举 $k$ ，此时有 $i_1、i_2、……i_m$ 都满足 $S_i=S_k$ ，那么它们对答案的贡献为： $\sum_{t=1}^m(k-1-i_t)=m*(k-1)-\sum_{t=1}^mi_t$ ，两个哈希表即可解决。

class Solution {
public:
    int countTriplets(vector<int>& arr) {
        int n=arr.size(),cur=0,ans=0;
        unordered_map<int,int> cnt,sum;
        cnt[0]=1;
        for(int i=1;i<=n;i++)
        {
            cur^=arr[i-1];
            ans+=cnt[cur]*(i-1)-sum[cur];
            ++cnt[cur];
            sum[cur]+=i;
        }
        return ans;
    }
};