题目
有一个正整数数组 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 * 104
1 <= arr[i] <= 109
1 <= queries.length <= 3 * 104
queries[i].length == 2
0 <= queries[i][0] <= queries[i][1] < arr.length
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/xor-queries-of-a-subarray
题解
方法
前缀异或
性质
- 任何数与自身异或等于0
- a^a=0
例如:(3^ 4 ^ 5 ^ 6 ^ 8)^ (3^ 4)= 5 ^ 6 ^ 8
- 任何数与0异或等于自身
- a^0=a
思路
一般想法:
- 对每个查询,计算数组中的对应下标范围内的元素的异或结果
每个查询的计算时间取决于查询对应的下标范围的长度。
如果数组arr的长度为n,数组 queries的长度为 m(即有m个查询),则最坏情况下每个查询都需要 O(n)的时间计算结果,总时间复杂度是 O(nm),会超出时间限制,因此必须优化。
由于有m个查询,对于每个查询都要计算结果,因此应该优化每个查询的计算时间。
理想情况下,每个查询的计算时间应该为O(1)。为了将每个查询的计算时间从O(n) 优化到O(1),需要计算数组的前缀异或。
例如:
求 query=[2,4]的查询结果result[2,4]=num2⨁num3⨁num4
因为 S5=(num0⨁num1)⨁(num2⨁num3⨁num4)
所以 S5=S2⨁result[2,4]
所以 result[2,4]=S5⨁S2
步骤
- 计算前缀异或数组 xors
- 计算每个查询的结果
第 i个查询的结果为 xors[queries[i][0]] ^ xors[queries[i][1]+1]。
代码
class Solution {
public int[] xorQueries(int[] arr, int[][] queries) {
//求前缀和
int n=arr.length;
//+1因为我们要插入一个0在数组的前面!
int [] xors=new int[n+1];
for(int i=0;i<n;i++){
xors[i+1]=xors[i]^arr[i];
}
//计算区间异或值
int m=queries.length;
int[] ans=new int[m];
for(int i=0;i<m;i++){
ans[i]=xors[queries[i][0]]^xors[queries[i][1]+1];
}
return ans;
}
}
复杂度分析
- 时间复杂度:O(n+m),其中 n 是数组 arr 的长度,m 是数组 queries 的长度。需要遍历数组arr 一次,计算前缀异或数组的每个元素值,然后对每个查询分别使用O(1) 的时间计算查询结果。
- 空间复杂度:O(n),其中n 是数组 arr 的长度。需要创建长度为 n+1 的前缀异或数组,注意返回值不计入空间复杂度。
参考
图片感谢力扣@小虎
https://leetcode-cn.com/problems/xor-queries-of-a-subarray/solution/xiao-ming-zuo-shu-xue-3-qian-zhui-yi-huo-3oaq/