C++ 中自右向左小於當前元素的個數


假設我們有一個數組 nums,我們需要找到另一個名為 count 的陣列,在這個 count 陣列中,count[i] 儲存 nums[i] 右側較小元素的數量。

因此,如果輸入類似於:[5,2,7,1],則結果將為 [2,1,1,0]。

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

  • 定義一個名為 update() 的方法,它將接收索引、陣列 bit 和 n 作為引數。

  • 當 index <= n 時,執行 -

    • 增加 bit[index] 的值。

    • index = index + (index) AND ( - index)

  • 定義一個名為 query() 的陣列,它將接收索引和 bits 陣列作為引數。

    • ans := 0

    • 當 index > 0 時,執行 -

      • ans = ans + bit[index]

      • index = index – (index AND - index)

    • 返回 ans

  • 從主方法中,執行以下操作 -

  • n := nums 的大小。

  • 定義一個大小為 n 的陣列 res。

  • 如果 (n 不為零) 為假,則返回 res。

  • maxx := 0,minn = nums [0]

  • 初始化 i := 1,當 i < n 時,將 i 增加 1,執行 -

    • minn := nums[i] 和 minn 中的最小值。

  • 初始化 i := 0,當 i < n 時,將 i 增加 1,執行 -

    • nums[i] := nums[i] - minn + 1

    • maxx := nums[i] 和 maxx 中的最大值。

  • 定義一個大小為 maxx + 1 的陣列 bit。

  • 初始化 i := n - 1,當 i >= 0 時,將 i 減小 1,執行 -

    • num = nums[i] – 1

    • x := 呼叫函式 query(num,bit)

    • res[i] := x

    • 呼叫函式 update(num + 1, bit, maxx)

  • 返回 res

示例

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

 線上演示

#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:
   void update(int index,vector <int>& bit, int n){
      while(index<=n){
         bit[index]++;
         index += (index)&(-index);
      }
   }
   int query(int index, vector <int> bit){
      int ans = 0;
      while(index>0){
         ans+=bit[index];
         index -= index&(-index);
      }
      return ans;
   }
   vector<int> countSmaller(vector<int>& nums) {
      int n = nums.size();
      vector <int> res(n);
      if(!n)return res;
      int maxx = 0;
      int minn = nums[0];
      for(int i =1;i<n;i++)minn = min(nums[i],minn);
      for(int i =0;i<n;i++){
         nums[i] = nums[i]-minn+1;
         maxx = max(nums[i],maxx);
      }
      vector <int> bit(maxx+1);
      for(int i =n-1;i>=0;i--){
         int num = nums[i]-1;
         int x = query(num,bit);
         res[i] = x;
         update(num+1,bit,maxx);
      }
      return res;
   }
};
main(){
   Solution ob;
   vector<int> v = {5,2,7,1};
   print_vector(ob.countSmaller(v));
}

輸入

[5,2,7,1]

輸出

[2, 1, 1, 0, ]

更新於: 2020年5月27日

301 次檢視

開啟您的 職業生涯

透過完成課程獲得認證

開始學習
廣告