力扣1310子数组异或查询

发表于 更新于 Changyan：

力扣1310子数组异或查询

问题

有一个正整数数组 arr，现给你一个对应的查询数组 queries，其中 queries[i] = [Li, Ri]。

对于每个查询 i，请你计算从 Li 到 Ri 的 XOR 值（即 arr[Li] xor arr[Li+1] xor … xor arr[Ri]）作为本次查询的结果。

并返回一个包含给定查询 queries 所有结果的数组。

示例 1：

输入：arr = [1,3,4,8], queries = [[0,1],[1,2],[0,3],[3,3]]
输出：[2,7,14,8]
解释：
数组中元素的二进制表示形式是：
1 = 0001
3 = 0011
4 = 0100
8 = 1000
查询的 XOR 值为：
[0,1] = 1 xor 3 = 2
[1,2] = 3 xor 4 = 7
[0,3] = 1 xor 3 xor 4 xor 8 = 14
[3,3] = 8

示例 2：

输入：arr = [4,8,2,10], queries = [[2,3],[1,3],[0,0],[0,3]]
输出：[8,0,4,4]

提示：

1 <= arr.length <= 3 * 10^4
1 <= arr[i] <= 10^9
1 <= queries.length <= 3 * 10^4
queries[i].length == 2
0 <= queries[i][0] <= queries[i][1] < arr.length

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/xor-queries-of-a-subarray

解答

前缀和的思想。使用暴力法会超时

1
2
3
4
5
6
7
8
9
10
11
class Solution:
    def xorQueries(self, arr: List[int], queries: List[List[int]]) -> List[int]:
        """XOR前缀和"""
        pre = [0]  # 前缀数组
        for i in range(len(arr)):
            pre.append(arr[i] ^ pre[-1])
        res = []
        for a, b in queries:
            res.append(pre[a] ^ pre[b + 1])

        return res

评论区神解答

1
2
3
4
class Solution:
    def xorQueries(self, arr: List[int], queries: List[List[int]]) -> List[int]:
        acc = list(accumulate([0] + arr, xor))
        return [acc[a] ^ acc[b + 1] for a, b in queries]