【数据结构-前缀异或和】1442. 形成两个异或相等数组的三元组数目

给你一个整数数组 arr 。

现需要从数组中取三个下标 i、j 和 k ,其中 (0 <= i < j <= k < arr.length) 。

a 和 b 定义如下:

a = arri ^ arri + 1 ^ ... ^ arrj - 1

b = arrj ^ arrj + 1 ^ ... ^ arrk

注意:^ 表示 按位异或 操作。

请返回能够令 a == b 成立的三元组 (i, j , k) 的数目。

示例 1:

输入:arr = 2,3,1,6,7

输出:4

解释:满足题意的三元组分别是 (0,1,2), (0,2,2), (2,3,4) 以及 (2,4,4)

示例 2:

输入:arr = 1,1,1,1,1

输出:10

示例 3:

输入:arr = 2,3

输出:0

示例 4:

输入:arr = 1,3,5,7,9

输出:3

示例 5:

输入:arr = 7,11,12,9,5,2,7,17,22

输出:8

前缀哈希

cpp 复制代码
class Solution {
public:
    int countTriplets(vector<int>& arr) {
        unordered_map<int, int> total, cnt;
        int s = 0, ans = 0;
        for(int k = 0; k < arr.size();k++){
            int val = arr[k];
            if(cnt.count(s ^ val)){
                ans += cnt[s ^ val] * k - total[s ^ val];
            }
            total[s] += k;
            cnt[s]++;
            s ^= val;
        }
        return ans;
    }
};

时间复杂度:O(n)。其中 n 是数组 arr 的长度。

空间复杂度:O(n)。我们需要使用 O(n) 的空间存储两个哈希表。

这道题的难点第一个是要找到前缀异或和直接的关系进行如上公式变换。

接下来最后可以得到以下这个式子:

首先第一个哈希表cnt是用来在遍历k,计算k+1的前缀异或和的时候,是否有对应的 i 的前缀异或和一样。cnt的键用来储存前缀异或和,值是对应的个数。每当有一个前缀异或和,则 j 可以在 从 i 到 k 之间任意一个位置,也就是 k - i 种情况。那么在遍历 k 的时候,就可以根据当前的前缀异或和来查找之前相同前缀异或和的个数cnts \^ val

以上的(i1 + i2 +...+ im)可以用哈希表total来记录所有I的和,最终结果就是
ans += cnt[s ^ val] * k - total[s ^ val];

最后遍历完arr返回ans即可

相关推荐
Tairitsu_H2 小时前
[LC优选算法#18] 前缀和 | 除⾃⾝以外数组的乘积 | 和为K的⼦数组 | 和可被K整除的⼦数组
算法·前缀和·哈希算法
凯瑟琳.奥古斯特2 小时前
力扣1008:前序重建BST
开发语言·c++·算法·leetcode·职场和发展
aqiu1111113 小时前
【算法日记 13】LeetCode 49. 字母异位词分组:哈希表的进阶“降维打击”
算法·leetcode·散列表
WL学习笔记3 小时前
链式队列(数据结构)
前端·数据结构·jquery
KaMeidebaby3 小时前
卡梅德生物技术快报|纳米抗体技术全套实操流程:AFB1 全合成文库淘选 + 分子对接定点突变参数手册
人工智能·python·tcp/ip·算法·机器学习
巴糖4 小时前
从做早餐理解 DAG 与拓扑排序:任务依赖图的核心逻辑
算法
2601_962851744 小时前
计算机毕业设计之基于YOLOV10的低光照目标检测算法研究与实现
大数据·算法·yolo·目标检测·信息可视化·cnn·课程设计
magicwt4 小时前
快手自动出价论文GAVE阅读笔记
算法
智者知已应修善业4 小时前
【P12159蓝桥杯数组翻转】2025-4-24
c语言·c++·经验分享·笔记·算法·蓝桥杯
tkevinjd5 小时前
力扣73矩阵置零
java·算法·面试·职场和发展