solution S239, S703

Author: xiao2shiqi

src/main/java/S239.java

@@ -0,0 +1,44 @@
+import java.util.Arrays;
+import java.util.LinkedList;
+
+/**
+ * LC#239：Sliding Window Maximum 滑动窗口最大值
+ * Link：https://leetcode-cn.com/problems/sliding-window-maximum/
+ * Solution：双端队列
+ * @author Phoenix on 2021/8/8.
+ */
+public class S239 {
+
+    public int[] maxSlidingWindow(int[] nums, int k) {
+        // 长度太短没必要执行代码
+        if(nums == null || nums.length < 2) return nums;
+        LinkedList<Integer> queue = new LinkedList<>();
+        // 算出结果集长度
+        int[] result = new int[nums.length - k + 1];
+        // 遍历 nums 数组
+        for (int i = 0; i < nums.length; i++) {
+            // 如果元素比较小，直接弹出
+            while (!queue.isEmpty() && nums[queue.peekLast()] <= nums[i]) {
+                queue.pollLast();
+            }
+            // 添加下标
+            queue.addLast(i);
+            // 判断队列首部是否有效
+            if(queue.peek() <= i-k) {
+                queue.poll();
+            }
+            // 当长度为 k 时，保存当前窗口最大值
+            if(i+1 >= k) {
+                result[i+1-k] = nums[queue.peek()];
+            }
+        }
+        return result;
+    }
+
+    public static void main(String[] args) {
+        int[] nums = {1,3,-1,-3,5,3,6,7};
+        int k = 3;
+        int[] ints = new S239().maxSlidingWindow(nums, k);
+        System.out.println("output:" + Arrays.toString(ints));
+    }
+}

src/main/java/S703.java

@@ -0,0 +1,54 @@
+import java.util.PriorityQueue;
+
+/**
+ * LC#703：数据流中的第 K 大元素
+ * Link：https://leetcode-cn.com/problems/kth-largest-element-in-a-stream/
+ * Solution：维护一个 MinHeap，保证堆的 size = K，堆顶为 K 元素，如果比 K 小则无法进入堆，如果比 K 大则自动淘汰最小元素，时间复杂度 O(1)
+ * @author Phoenix on 2021/8/8.
+ */
+public class S703 {
+
+    // 直接使用 Java PriorityQueue（默认小顶堆）
+    final PriorityQueue<Integer> q;
+    final int k;
+
+    public S703(int k, int[] a) {
+        this.k = k;
+        q = new PriorityQueue<>(k);
+        // 加入堆顶
+        for (int n : a) {
+             add(n);
+        }
+    }
+
+    public int add(int n) {
+        if (q.size() < k) {
+            q.offer(n);
+        }else if (q.peek() < n) {
+            q.poll();
+            q.offer(n);
+        }
+        return q.peek();
+    }
+
+    @Override
+    public String toString() {
+        return "S703{" +
+                "q=" + q +
+                ", k=" + k +
+                '}';
+    }
+
+    public static void main(String[] args) {
+        int[] a = {4, 5, 8, 2};
+        S703 s = new S703(3,  a);
+        s.add(3);
+        s.add(5);
+        s.add(10);
+        s.add(9);
+        s.add(4);
+
+        System.out.println(s);
+    }
+
+}