C++ 中子数组的异或查询
假设我们有一个包含正整数的数组 arr 和一个查询数组 queries,其中 queries[i] = [Li, Ri],对于每个查询 i,计算从 Li 到 Ri 的元素的异或(即,arr[Li] XOR arr[Li+1] XOR ... XOR arr[Ri])。我们必须找到一个包含给定查询结果的数组。所以如果输入像 - [1,3,4,8],并且查询像 [[0,1],[1,2],[0,3],[3,3]],那么结果将是 [2,7,14,8]。这是因为数组中元素的二进制表示为 - 1 = 0001,3 = 0011,4 = 0100 和 8 = 1000。然后查询的异或值为 - [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
为了解决这个问题,我们将遵循以下步骤 -
- n := arr 的大小
- 定义一个名为 pre 的数组,大小为 n + 1,然后填充 pre,使得 pre[i] := pre[i – 1] XOR arr[i – 1]
- 定义另一个数组 ans
- 对于 i 从 0 到查询数 - 1
- l := queries[i, 0],r := queries[i, 1]
- 将 l 和 r 加 1
- 将 pre[r] XOR pre[l - 1] 插入 ans
- 返回 ans
示例(C++)
让我们看看以下实现,以便更好地理解 -
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: vector<int> xorQueries(vector<int>& arr, vector<vector<int>>& queries) { int n = arr.size(); vector <int> pre(n + 1); for(int i = 1; i <=n; i++){ pre[i] = pre[i - 1] ^ arr[i - 1]; } vector <int> ans; for(int i = 0; i < queries.size(); i++){ int l = queries[i][0]; int r = queries[i][1]; l++; r++; ans.push_back(pre[r] ^ pre[l - 1]); } return ans; } }; main(){ vector<int> v = {1,3,4,8}; vector<vector<int>> v1 = {{0,1},{1,2},{0,3},{3,3}}; Solution ob; print_vector(ob.xorQueries(v, v1)); }
输入
[1,3,4,8] [[0,1],[1,2],[0,3],[3,3]]
输出
[2,7,14,8]
广告