形成两个异或相等数组的三元组数目(前缀和)

在这里插入图片描述
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;
    }
}


  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值