首頁  >  文章  >  後端開發  >  PHP如何計算資料流中的第K大的元素

PHP如何計算資料流中的第K大的元素

醉折花枝作酒筹
醉折花枝作酒筹轉載
2021-07-09 15:59:462172瀏覽

利用最小堆的性質,此最小堆的根結點一定是所有結點中最小的。所以,我們只需要維護K個元素大小的最小堆。只要是大於最小堆的根結點的值,就移除該根結點的值,把該值插入最小堆中。

PHP如何計算資料流中的第K大的元素

設計一個找到資料流中第K大元素的類別(class)。注意是排序後的第K大元素,不是第K個不同的元素。

你的 KthLargest 類別需要同時接收整數 k 和整數陣列nums 的建構器,它包含資料流中的初始元素。每次呼叫 KthLargest.add,返回目前資料流中第K大的元素。

範例:

int k = 3;
int[] arr = [4,5,8,2];
KthLargest kthLargest = new KthLargest(3, arr);
kthLargest.add(3);   // returns 4
kthLargest.add(5);   // returns 5
kthLargest.add(10);  // returns 5
kthLargest.add(9);   // returns 8
kthLargest.add(4);   // returns 8

說明: 你可以假設 nums 的長度≥ k-1 且k ≥ 1。

理解題意

最開始沒理解這題,讀了幾遍明白了,k 是指定的位置,然後一組數,按從大到小排列,然後返回第k 個元素,最基礎考慮是每次add 都需要一次排序,然後重新查找第k 個元素的值,例如5,3,6,2 四個數,倒敘之後是6, 5,3,2,則k = 2 的情況下,返回5,然後add 1的情況,追加在最末尾,不影響結果,add 7 的情況,追加在最開頭,返回值變成6。

解題想法

直接使用最小堆,堆的大小為k,這樣保證空間佔用最小,最小堆的根節點是就是最小值,也是我們想要的結果。 PHP 的 SPL 標準函式庫是有最小堆這個函式庫,直接在程式碼中繼承 SplMinHeap 。

PHP 實作

class KthLargest extends SplMinHeap {
    /** 
    * @param Integer $k 
    * @param Integer[] $nums 
    */
    static $nums;
    public $k;
    function __construct($k, $nums) {
        $this->k = $k;
        // 遍历初始化数组,分别插入堆中
        foreach ($nums as $v) {
            $this->add($v);
        }
    }
  
    /** 
    * @param Integer $val 
    * @return Integer 
    */
    function add($val) {
       // 维持堆的大小为k,当堆还未满时,插入数据。
        if ($this->count() < $this->k) {
            $this->insert($val);
        } elseif ($this->top() < $val) {
        // 当堆满的时候,比较要插入元素和堆顶元素大小。大于堆顶的插入。堆顶移除。
            $this->extract();
            $this->insert($val);
        }
        return $this->top();
    }}
    /** 
    * Your KthLargest object will be instantiated and called as such: 
    * $obj = KthLargest($k, $nums); 
    * $ret_1 = $obj->add($val); 
    */

推薦學習:php影片教學

#

以上是PHP如何計算資料流中的第K大的元素的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文轉載於:hxd.life。如有侵權,請聯絡admin@php.cn刪除