1. 程式人生 > >[Leetcode215] 陣列中的第K個最大元素

[Leetcode215] 陣列中的第K個最大元素

陣列中的第K個最大元素

目前能想到的就是先排序在搜尋,因為為了找到第K大的數必須要所有數都比較。所以這道題關鍵點在排序和搜尋上,前者涉及演算法,後者涉及資料結構。

我用了內建函式,應該算作弊了哈哈哈

python:

class Solution(object):
    def findKthLargest(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        nums.sort()
        return nums[-k]

C++: 

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        return nums[nums.size()-k];
    }
};