LeetCode.378. 有序矩阵中第K小的元素

# 该题的标签为 1.堆    2. 二分查找

1. 首先我们用堆的思想来解决该问题

    1.1 该矩阵的特点是 n*n 且每行和每列都是有序的

   

    1.2 以上图矩阵为例，我们从行的角度去考虑（也可以从列）

         每行的第一个为最小值，我们把1，10，12 放入最小堆

    

    1.3 接下来我们从最小堆中，取出堆顶元素。同时，对于取出的元素，

         我们要判断，该元素存不存在下一个元素，如果存在则，把该元素

        的下一个元素入堆。

       

       如上图，取出的1，进入的是5. 此时堆内还有 5，10，12

    1.4 继续 1.3 当出堆次数为k-1 次时停止。

          核心思想就是，每次取出的是当前堆中的最小元素，此时，要

     尽快把大于该元素的值放入堆。从行的角度来看，大于该元素的值，

     就是行内的下一个元素。 

        

class Solution {
    static class Node implements Comparable<Node> {
        private int x, y, value;

        public Node(int x, int y, int value) {
            this.x = x;
            this.y = y;
            this.value = value;
        }

        @Override
        public int compareTo(Node node) {
            return this.value - node.value;// 从小到大
        }

        public int getX() {
            return x;
        }

        public int getY() {
            return y;
        }

        public int getValue() {
            return value;
        }
    }

    public int kthSmallest(int[][] matrix, int k) {
        // 使用堆
        PriorityQueue<Node> pQValues = new PriorityQueue<>();
        int n = matrix.length;
        // 把每一行的第一个元素入堆，每一行的第一个都是该行的最小值
        for (int x = 0; x < n; x++) {
            Node _node = new Node(x, 0, matrix[x][0]);
            pQValues.offer(_node);
        }
        // 要找到第k小的数,需要把前k-1个最小的数丢弃掉
        for (int i = 0; i < k - 1; i++) {
            Node _node = pQValues.poll();// 从队列中取出一个，一共需要执行k-1次
            if (_node.getY() == n - 1) {
                // 如果取到一行中的最后一个
                continue;
            } else {
                // 取该数的下一个
                Node _n = new Node(_node.getX(), _node.getY() + 1, matrix[_node.getX()][_node.getY() + 1]);
                pQValues.offer(_n);
            }
        }
        // 此时在最小堆中的第一个元素就是第k个最小值
        return pQValues.peek().getValue();
    }
}

2. 使用二分法，后续补充