原贴地址:http://blog.leanote.com/post/dawnmagnet/0bb52ca14c01
题目
有一个正整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析
又是一道异或的题目。
我们再强调一下,异或的最重要的性质就是两个相同的数字异或是0
接下来看一下我们的这个数据量,不太适合暴力,如果按照题目的说法进行遍历,复杂度就是
O
(
a
r
r
.
l
e
n
g
t
h
∗
q
u
e
r
i
e
s
.
l
e
n
g
t
h
)
O(arr.length*queries.length)
O(arr.length∗queries.length),达到了
9
∗
1
0
8
9*10^8
9∗108复杂度,这种已经很大了,很危险,很有可能过不了。这也是对于所有题目的一个小提示:如果复杂度达到
1
0
9
10^9
109,就过不了了。
这个题目既然如此危险,不到万不得已就不写暴力算法。
此处说明一下优化的手段,一般来说O(n^2)的算法只能往O(n)和O(nlgn)进行优化,但这里我们观察了一下也没什么可以二分的办法,所以我们就用O(n)进行思考,既然是O(n)那么每一次查询都应该是O(1)的复杂度,有可能吗?
这里就要介绍一下前缀和的思想。
我们先思考一个跟本问题很像的一个问题,就是我们将本题中的异或换成加法。我们给入一个查询,查询的是数组中间某一段的和。为了在O(1)的时间应对每次查询,我们新建一个前缀和数组,称为prefix
prefix中的下标 | nums中的下标 |
---|---|
0 | n0 |
1 | n0+n1 |
2 | n0+n1+n2 |
3 | n0+n1+n2+n3 |
4 | n0+n1+n2+n3+n4 |
… | … |
如果我们获得了前缀和数组,对于每次查询都能在O(1)的时间返回,因为如果我们想要获得1至3之间的值(包含3),我们就用**prefix[3]-prefix[0]**就获得了1至3之间的值的和。
我们再仔细思考一下为什么是加法就可以这么做。以及换成异或之后我们应该如何做。
性质|加法+|异或^
–|--
逆操作|减法-|异或^
一顺一逆的结果|0+n-n=0|0nn=0
现在我们可以知道为什么上述操作中我们会用prefix[3]减去prefix[0],因为减法是加法的逆操作。而正因为一顺一逆的结果是0,这个前缀和才有它存在的意义。
我们仿照着加法设计出来我们的异或前缀和表
prefix中的下标 | nums中的下标 |
---|---|
0 | n0 |
1 | n0^n1 |
2 | n0n1n2 |
3 | n0n1n2^n3 |
4 | n0n1n2n3n4 |
… | … |
同样的,我们如果想获得nums中从1异或到3的值,我们只需要获取prefix[3]^prefix[0],也是一个O(1)的操作。
最后说一下为什么选择前缀和,因为大量查询,查询的复杂度完全将生成前缀和的复杂度掩盖了(数量级碾压),因为处理出前缀和是O(n)的操作,而查询在优化后也是大于等于O(n)的级别的,我们才有使用前缀和的意义。
C++代码
class Solution {
public:
vector<int> xorQueries(vector<int>& arr, vector<vector<int>>& queries) {
int n = arr.size();
vector<int> prefix = {0};
for (int i = 0; i < n; ++i) prefix.push_back(prefix[i] ^ arr[i]);
vector<int> res;
for (auto & query : queries) res.push_back(prefix[query[1] + 1] ^ prefix[query[0]]);
return res;
}
};
Rust代码
impl Solution {
pub fn xor_queries(arr: Vec<i32>, queries: Vec<Vec<i32>>) -> Vec<i32> {
let n = arr.len();
let mut prefix = vec![0];
for i in 0..n {
prefix.push(prefix[i] ^ arr[i]);
}
let mut res = vec![];
for query in queries {
res.push(prefix[query[1] as usize + 1] ^ prefix[query[0] as usize])
}
res
}
}