add q215/q347

Author: yuanguangxin

.idea/workspace.xml

@@ -48,10 +48,14 @@
 * [q54_螺旋矩阵](/src/数组操作/q54_螺旋矩阵)
 * [q73_矩阵置零](/src/数组操作/q73_矩阵置零)
 
-### 栈操作
+### 栈相关
+
+* [q20_有效的括号](/src/栈相关/q20_有效的括号)
+* [q224_基本计算器](/src/栈相关/q224_基本计算器)
+
+### 堆相关
+
 
-* [q20_有效的括号](/src/栈操作/q20_有效的括号)
-* [q224_基本计算器](/src/栈操作/q224_基本计算器)
 
 ### 区间合并
 

README.md

@@ -0,0 +1,23 @@
+package 堆相关.q215_数组中的第K个最大元素;
+
+import java.util.PriorityQueue;
+
+/**
+ * 利用大根堆实现 o(n*log(k))
+ */
+public class Solution {
+
+    public int findKthLargest(int[] nums, int k) {
+        PriorityQueue<Integer> heap =
+                new PriorityQueue<>((n1, n2) -> n1 - n2);
+
+        for (int n: nums) {
+            heap.add(n);
+            if (heap.size() > k){
+                heap.poll();
+            }
+        }
+
+        return heap.poll();
+    }
+}

src/堆相关/q215_数组中的第K个最大元素/Solution.java

@@ -0,0 +1,32 @@
+package 堆相关.q347_前K个高频元素;
+
+import java.util.*;
+
+/**
+ * 利用大根堆（PriorityQueue）实现 o(n*log(k))
+ */
+class Solution {
+    public List<Integer> topKFrequent(int[] nums, int k) {
+
+        HashMap<Integer, Integer> count = new HashMap<>();
+        for (int n : nums) {
+            count.put(n, count.getOrDefault(n, 0) + 1);
+        }
+
+        PriorityQueue<Integer> heap = new PriorityQueue<>(Comparator.comparingInt(count::get));
+
+        for (int n : count.keySet()) {
+            heap.add(n);
+            if (heap.size() > k) {
+                heap.poll();
+            }
+        }
+
+        List<Integer> topK = new LinkedList<>();
+        while (!heap.isEmpty()) {
+            topK.add(heap.poll());
+        }
+        Collections.reverse(topK);
+        return topK;
+    }
+}

src/堆相关/q347_前K个高频元素/Solution.java

@@ -1,4 +1,4 @@
-package 栈操作.q20_有效的括号;
+package 栈相关.q20_有效的括号;
 
 import java.util.Stack;
 

src/栈操作/q20_有效的括号/Solution.java src/栈相关/q20_有效的括号/Solution.java

@@ -1,4 +1,4 @@
-package 栈操作.q224_基本计算器.f1;
+package 栈相关.q224_基本计算器.f1;
 
 import java.util.Stack;
 

src/栈操作/q224_基本计算器/f1/Solution.java src/栈相关/q224_基本计算器/f1/Solution.java

@@ -1,4 +1,4 @@
-package 栈操作.q224_基本计算器.f2;
+package 栈相关.q224_基本计算器.f2;
 
 import java.util.Stack;