復(fù)雜度為n的topk問題

思想:類似快排的分治思想,把數(shù)組每次都按照pivot分左右兩邊,當pivotIdx等于我們的k-1時,這個pivot就是我們要找的topK,否則則從左或右數(shù)組再次遍歷。直到數(shù)組只剩下一個元素為止。

<html>
    <script>
        //resolve topK

        function topk(arr,k){
            //終止條件,數(shù)組只剩下一個時
            if(arr.length===1){
                return arr[0]
            }
            let pivot = arr[arr.length-1]
            let left = []
            let right = []
            for(let i=0;i<arr.length-1;i++){
                //小于等于為了保證排序穩(wěn)定
                if(arr[i]<=pivot){
                    left.push(arr[i])
                }else{
                    right.push(arr[i])
                }
            }
            //左數(shù)組的長度即為pivot的Index
            let pivotIdx = left.length
            if(pivotIdx === k-1 ){
                return pivot
            }else if(left.length>=k){
                //這里記得要return這個函數(shù),否則無法接受到return值
                return topk(left,k)
            }else{
                return topk(right,k-left.length-1)
            }
        }


        let k = topk([4,2,5,12,-1,2,5,6,-5,3],3)
        console.log('k',k);//2

        /*
        時間復(fù)雜度分析
            每次都要遍歷數(shù)組的一半(左半和右半),雖然左右數(shù)組大小可能不同,但都是類似分半了
            所以遍歷次數(shù)為n,n/2,n/4...1
            很顯然是一個等比數(shù)組,
            s=n+n/2+n/4+...1
            用點高中的小技巧
            2s=2n+n+n/2+..2
            2s-s=2n-1
            s=2n-1
            所以這個時間復(fù)雜度是n
        */
        
    </script>
</html>
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容