找出數組中第 k 大的數字及其出現次數

這是前端面試過程當中遇到的一道算法題,雖然說難度不大,可是也有些細節的地方須要仔細考慮。javascript

好比說數組 [1, 2, 4, 4, 3, 5],第 2 大的數字是 4,出現了 2 次。下面以這個爲例展開算法的講解。前端

先說說大致的思路,而後再考慮一些細節性的問題。java

  • 既然涉及到數字大小的問題,那就要對給定數組進行排序,題目要求「第 k 大」的數字,故選擇降序的方式更有利於後面的查找;面試

  • 重點來了,須要遍歷數組,對當前遍歷到的數組元素的大小排名和 k 值進行比較,若排名大於 k 值則結束循環;若是排名等於 k 值,則將該數組元素記爲第 k 大的數字,且次數加一;算法

  • 那如何肯定遍歷的當前數組元素的大小排名呢?咱們先將數組的第一項定爲第 1 大的數字,而後比較第一項和第二項是否相等,若相等則第二項也爲第 1 大;若是不相等則數組第二項爲第 2 大的數字。依次類推,可獲知每次遍歷的數組元素的大小排名。數組

總體的思路講完了,如今看看細節性的東西——邊界判斷測試

  • 給定的 k 值是否大於或等於 1(保證沒有第 0 大的數字),且 k 值小於或等於數組的長度(由於數組經降序排序後,值最小的數組元素的大小排名值不可能比數組長度值更大)。這條判斷隱含了數組不能爲空的條件。ui

  • 遍歷結束時,k 值不能超出值最小的數組元素的大小排名,若超出了,說明數組中不存在這樣的數字。spa

根據以上思路和細節考慮寫出的算法以下:code

function findKthNum (arr, k) {
  var len = arr.length;
  // 對給定的 k 值進行判斷,確保 len >= k >= 1
  if (k > len || k < 1) {
    console.log("There doesn't exists the number you want !");
    return;
  }
  // 得到數組的副本
  var _arr = arr.slice();
  // 遍歷數組時,當前數組元素的大小排名
  var rank = 1;
  // 第 k 大的數字
  var num = 0;
  // 第 k 大數字的出現次數
  var count = 0;
  // 對 _arr 進行降序排序
  _arr.sort(function (a, b) {
    return b - a;
  });
  for (var i = 0; i < len; i++) {
    var j = i + 1;
    // 對當前數組元素的大小排名和 k 值進行比較,若排名大
    // 於 k 值則結束循環;若是排名等於 k 值,則將該數組
    // 元素記爲第 k 大的數字,且次數加一。
    if (rank > k) {
      break;
    } else if (rank == k) {
      num = _arr[i];
      count++;
    }
    // 若當前遍歷的數組項與下一項數字不相等,則說明下一個
    // 數字的排名值比當前遍歷數字的排名恰好大 1
    if ((j < len) && (_arr[i] != _arr[j])) {
      rank++;
    }
  }
  // 遍歷結束時,若最後遍歷的數組元素的大小排名小於給定的 k 值,則說
  // 明數組中沒有第 k 大的數字,即 k 值超出了全部數組元素的大小排名。
  if (rank < k) {
    console.log("There doesn't exists the number you want !");
  } else {
    console.log('第' + k + '大的數字:' + num, '出現次數:' + count);
  }
}
複製代碼

最後給出一些測試用例:

// 正常狀況,findKthNum(arr1, 2)
var arr1 = [1, 2, 4, 4, 3, 5];

// 數組爲空,findKthNum(arr2, 2)
var arr2 = [];

// k 值小於 1,findKthNum(arr3, 0)
var arr3 = [1, 2, 4, 4, 3, 5];

// k 值大於數組長度,findKthNum(arr4, 7)
var arr4 = [1, 2, 4, 4, 3, 5];

// k 值超出了全部數組元素的大小排名,findKthNum(arr5, 6)
var arr5 = [1, 2, 4, 4, 3, 5];
複製代碼
相關文章
相關標籤/搜索