C++ 中的雙倍對陣列


假設我們有一個長度為偶數的整數陣列 A,現在我們必須在只有當可以對其進行重新排序以使 A[2 * i + 1] = 2 * A[2 * i] 對每個 0 <= i < len(A) / 2 成立時才返回 true。所以如果輸入像 [3,1,3,6],則結果將為 false,而 [4,-2,2,-4] 將返回 true。

為了解決這個問題,我們將遵循以下步驟 -

  • 建立一個對映 m,n := A 的大小,將 A 中每個元素的頻率儲存到對映 m 中

  • cnt := A 的大小

  • 對於對映中的每個鍵值對 kv

    • 如果 m[kv 的鍵] > 0,則

      • 如果 m[kv 的鍵] 不為 0 且 m[kv 的鍵 * 2] > 0

        • x := m[kv 的鍵] 和 m[kv 的鍵 * 2] 的最小值

        • cnt := cnt – (x * 2)

        • 將 m[kv 的鍵 * 2] 減少 x

        • 將 m[kv 的鍵] 減少 x

      • 否則當 kv 的鍵 = 0 時,則

        • cnt := cnt – m[kv 的鍵]

        • m[kv 的鍵] := 0

  • 當 cnt 不為零時返回 false,否則返回 true

讓我們看看以下實現以獲得更好的理解 -

示例

 現場演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   bool canReorderDoubled(vector<int>& A) {
      map <int, int> m;
      int n = A.size();
      for(int i = 0; i < n; i++){
         m[A[i]]++;
      }
      int cnt = A.size();
      map <int, int> :: iterator it = m.begin();
      while(it != m.end()){
         if(m[it->first] > 0){
            if(it->first != 0 && m[it->first * 2] > 0){
               int x = min(m[it->first], m[it->first * 2]);
               cnt -= (x * 2);
               m[it->first * 2] -= x;
               m[it->first] -= x;
            }else if(it->first == 0){
               cnt -= m[it->first];
               m[it->first] = 0;
            }
         }
         it++;
      }
      return !cnt;
   }
};
main(){
   vector<int> v1 = {3,1,3,6};
   Solution ob;
   cout << (ob.canReorderDoubled(v1)) << endl;
   v1 = {4,-2,2,-4};
   cout << (ob.canReorderDoubled(v1));
}

輸入

[3,1,3,6]
[4,-2,2,-4]

輸出

0
1

更新於: 2020-04-30

212 次檢視

開啟您的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.