题目
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
解题思路
- 首先需要做一个从0到i元素异或后的结果。
- 我们想计算 D, E 的结果,那可以用
map[E] ^ map[C] = A^B^C^D^E ^ A^B^C = A^A^B^B^C^C^D^E = 0^0^0^D^E = D ^ E
就可以很快算出结果了。 - 通过这个公式来进行求解。
Code
class Solution:
def xorQueries(self, arr: List[int], queries: List[List[int]]) -> List[int]:
ls = [0]
for i in arr:
ls.append(ls[-1] ^ i)
ls.remove(0) # 删除最开始添加的0
res = []
for l, r in queries:
if l == 0:
res.append(ls[r])
else:
# 通过 R 和 L-1 计算结果
res.append(ls[l-1] ^ ls[r])
return res