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 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.