C++ 中乘法表中的第 K 小數


假設我們知道一個乘法表。但是我們能否快速找到乘法表中的第 k 個最小數呢?如果我們有一個 m * n 乘法表,高度為 m,長度為 n,以及一個正整數 k,我們需要找到該表中第 k 個最小數。

例如,如果 m = 3,n = 3,k = 6,則輸出為 4。這是因為乘法表如下:


123
1123
2246
3369

第 6 小的元素是 4,因為 [1,2,2,3,3,4,6,6,9]

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

  • 定義一個函式 ok(),它將接收 m、n、x 作為引數。
  • ret := 0
  • for i := 1 to n do −
    • temp := min(x / i, m)
    • ret := ret + temp
  • return ret
  • 在主方法中,執行以下操作:
  • ret := -1, low := 1, high := m * n
  • while low <= high do −
    • mid := low + (high - low) / 2
    • cnt := ok(m, n, mid)
    • if cnt >= k then −
      • high := mid - 1
      • ret := mid
    • 否則
      • low := mid + 1
  • return ret

讓我們看下面的實現來更好地理解:

示例

線上演示

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   int ok(int m, int n, int x){
      int ret = 0;
      for(int i = 1; i <= n; i++){
         int temp = min(x / i, m);
         ret += temp;
      }
      return ret;
   }
   int findKthNumber(int m, int n, int k) {
      int ret = -1;
      int low = 1;
      int high = m * n ;
      while(low <= high){
         int mid = low + (high - low)/ 2;
         int cnt = ok(m, n, mid);
         if(cnt >= k){
            high = mid - 1;
            ret = mid;
         }else low = mid + 1;
      }
      return ret;
   }
};
main(){
   Solution ob;
   cout << (ob.findKthNumber(3,3,6));
}

輸入

“2*”

輸出

4

更新於:2020年6月1日

268 次瀏覽

開啟你的職業生涯

完成課程後獲得認證

開始學習
廣告
© . All rights reserved.