leetcode.com 215. Kth Largest Element in an Array

문제 링크: https://leetcode.com/problems/kth-largest-element-in-an-array

Author: skysign

leetcode.com 215. Kth Largest Element in an Array v2/main.py

@@ -0,0 +1,17 @@
+import sys
+from typing import List
+import heapq
+
+
+class Solution:
+    def findKthLargest(self, nums: List[int], k: int) -> int:
+        heap = []
+        answer = sys.maxsize
+
+        for n in nums:
+            heapq.heappush(heap, -n)
+
+        for _ in range(k):
+            answer = heapq.heappop(heap)
+
+        return -answer

leetcode.com 215. Kth Largest Element in an Array v2/test_main.py

@@ -0,0 +1,16 @@
+from unittest import TestCase
+from main import Solution
+
+
+class TestSolution(TestCase):
+    def test1_find_kth_largest(self):
+        sln = Solution()
+        self.assertEqual(5, sln.findKthLargest([3, 2, 1, 5, 6, 4], 2))
+
+    def test2_find_kth_largest(self):
+        sln = Solution()
+        self.assertEqual(4, sln.findKthLargest([3, 2, 3, 1, 2, 4, 5, 5, 6], 4))
+
+    def test5_find_kth_largest(self):
+        sln = Solution()
+        self.assertEqual(99, sln.findKthLargest([99, 99], 1))