题目
有一个正整数数组 arr,现给你一个对应的查询数组 queries,其中 queries[i] = [Li, Ri]。
对于每个查询 i,请你计算从 Li 到 Ri 的 XOR 值(即 arr[Li] xor arr[Li+1] xor … xor arr[Ri])作为本次查询的结果。
并返回一个包含给定查询 queries 所有结果的数组。
提示:
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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
看到这题我就知道要存数据了,不然时间是个大问题。(这几题可以出异或运算合集了,哪天统计一下。
先用普通方法写了一下。
答案
class Solution {
public:
vector<int> xorQueries(vector<int>& arr, vector<vector<int>>& queries) {
//对queries中的每个数组,计算arr下标从l到r的异或值
//如果数据量大可以使用map存储结果,具体怎么设计之后再看
//运行完超出时间限制了
int numofquery = queries.size();
vector<int> results(numofquery);
for(int i=0;i<numofquery;i++){
int left = queries[i][0];
int right = queries[i][1];
results[i] = xorquery(arr,left,right);
}
return results;
}
int xorquery(vector<int>& arr,int l,int r){
int result=0;
for(int i = l;i<=r;i++){
result = result ^ arr[i];
}
return result;
}
};
复杂度分析:
时间复杂度:如果数组arr 的长度为 n,数组queries 的长度为 m(即有 m 个查询),则最坏情况下每个查询都需要 O(n)的时间计算结果,总时间复杂度是 O(nm)
空间复杂度:O(1)。
改进
没写出来之前不小心看到了题解。因此知道了也是利用异或的性质。
a[i]^…^a[j] = (a[0]^…^a[i-1])^(a[0]^…a[i-i])^(a[i]^…^a[j])
class Solution {
public:
vector<int> xorQueries(vector<int>& arr, vector<vector<int>>& queries) {
//对queries中的每个数组,计算arr下标从l到r的异或值
//indexarr[i]是arr前i个数的异或值
//a[i]^...^a[j] = (a[0]^...^a[i-1])^(a[0]^...a[i-i])^(a[i]^...^a[j])
int numofquery = queries.size();
int numofarr = arr.size();
vector<int> indexarr(numofarr+1,arr[0]);
for(int i=1;i<numofarr;i++){
indexarr[i] = arr[i]^indexarr[i-1];
}
vector<int> results(numofquery);
for(int i=0;i<numofquery;i++){
int left = queries[i][0];
int right = queries[i][1];
if(left==0) results[i] = indexarr[right];
else results[i] = indexarr[left-1]^indexarr[right];
}
return results;
}
};
复杂度分析:
时间复杂度:O(n).
空间复杂度:O(n)。