数组中的第k个最大元素—leetcode215
生活随笔
收集整理的這篇文章主要介紹了
数组中的第k个最大元素—leetcode215
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
在未排序的數組中找到第 k 個最大的元素。請注意,你需要找的是數組排序后的第 k 個最大的元素,而不是第 k 個不同的元素。
示例 1:
輸入: [3,2,1,5,6,4] 和 k = 2
輸出: 5
示例?2:
輸入: [3,2,3,1,2,4,5,5,6] 和 k = 4
輸出: 4
說明:
你可以假設 k 總是有效的,且 1 ≤ k ≤ 數組的長度。
?解法一:冒泡排序,從后往前排,復雜度O(kn)
class Solution { public:int findKthLargest(vector<int>& nums, int k) {int n = nums.size();if(n==0){return 0;}int start = 0;while(start<k){for(int i=n-1;i>start;--i){if(nums[i]>nums[i-1]){int temp = nums[i-1];nums[i-1] = nums[i];nums[i] = temp;}}start++;}return nums[k-1];} };解法二:小頂堆,復雜度O(nlogk)
class Solution { public:int findKthLargest(vector<int>& nums, int k) {int N=nums.size();build_Ksmallstack(nums,k);for(int index=k;index<N;++index){if(nums[index]>nums[0]){swap(nums,index,0);smallstack(nums, k, 0);}}return nums[0];}void build_Ksmallstack(vector<int>& nums, int k){int noTreeNodes = (k-1)/2;for(int i=noTreeNodes;i>=0;--i){smallstack(nums,k,i);}return;}void smallstack(vector<int>& nums, int k, int index){int min = index;while(true){if(2*index+1<k && nums[2*index+1]<nums[index]){min = 2*index+1;}if(2*index+2<k && nums[2*index+2]<nums[min]){min = 2*index+2;}if(min==index){break;}swap(nums,min,index);index = min;}}void swap(vector<int>& nums,int min,int index){int temp = nums[min];nums[min] = nums[index];nums[index] = temp;return;} };?
總結
以上是生活随笔為你收集整理的数组中的第k个最大元素—leetcode215的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 二叉树的最近公共祖先—leetcode2
- 下一篇: 最长上升子序列—leetcode300