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]

更新于: 2020年4月30日

207 次查看

开启你的 职业生涯

通过完成课程获得认证

开始学习
广告