leetcode_378. 有序矩阵中第K小的元素_373. 查找和最小的K对数字

总结

• 在有序中找元素，用二分法；
• 在有序数组中第k小的元素，类比于合并m个链表，将每行看成一个链表，共m行，建立一个大小为m的堆，将元素从小到大放进堆中，弹出的第k个元素，就是第k小的元素
• 查找最小的k对数字是一样的。

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

给定一个 n x n 矩阵，其中每行和每列元素均按升序排序，找到矩阵中第 k 小的元素。
请注意，它是排序后的第 k 小元素，而不是第 k 个不同的元素。

示例：

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

返回 13。

思路

class Solution {
    public int kthSmallest(int[][] matrix, int k) {

        if(matrix == null || matrix.length == 0) return -1;
        if(matrix[0] == null || matrix[0].length == 0) return -1;

        int m = matrix.length, n = matrix[0].length;

        Queue<int[]> queue = new PriorityQueue<int[]>((array1, array2) -> array1[0] - array2[0]);

        for(int i = 0; i < m; i++) {
            queue.offer(new int[]{matrix[i][0], i, 0});
        }

        for(int i = 0; i < k - 1; i++) {
            int[] array = queue.poll();
            if(array[2] != n - 1) {
                queue.offer(new int[] {matrix[array[1]][array[2] + 1], array[1], array[2] + 1});
            }
        } 

        return queue.poll()[0];
    }
}

题目 373. 查找和最小的K对数字

给定两个以升序排列的整形数组 nums1 和 nums2, 以及一个整数 k。

定义一对值 (u,v)，其中第一个元素来自 nums1，第二个元素来自 nums2。

找到和最小的 k 对数字 (u1,v1), (u2,v2) … (uk,vk)。

示例 1:

输入: nums1 = [1,7,11], nums2 = [2,4,6], k = 3
输出: [1,2],[1,4],[1,6]
解释: 返回序列中的前 3 对数：
[1,2],[1,4],[1,6],[7,2],[7,4],[11,2],[7,6],[11,4],[11,6]
示例 2:

输入: nums1 = [1,1,2], nums2 = [1,2,3], k = 2
输出: [1,1],[1,1]
解释: 返回序列中的前 2 对数：
[1,1],[1,1],[1,2],[2,1],[1,2],[2,2],[1,3],[1,3],[2,3]
示例 3:

输入: nums1 = [1,2], nums2 = [3], k = 3
输出: [1,3],[2,3]
解释: 也可能序列中所有的数对都被返回:[1,3],[2,3]

思路

和合并k个链表相似
最小堆，
取出k-1个元素，第k个最小元素就是堆顶元素了

class Solution {
    public List<List<Integer>> kSmallestPairs(int[] nums1, int[] nums2, int k) {
        List<List<Integer>> res = new ArrayList(k);

        if(nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0) return res;

        int m = nums1.length, n = nums2.length;

        Queue<int[]> queue = new PriorityQueue<int[]>
                    ((array1, array2) -> array1[0] - array2[0]);

        for(int i = 0; i < m; i++) {
            queue.offer(new int[]{nums1[i] + nums2[0], i, 0});
        }

        for(int i = 0; i < k; i++) {
            int[] array = queue.poll();
            if(array == null) {
                break;
            }
            res.add(Arrays.asList(nums1[array[1]], nums2[array[2]]));
            if(array[2] != n - 1) {
                queue.offer(new int[]{nums1[array[1]] + nums2[array[2] + 1],
                                                         array[1], array[2]+1}); 
            }
        }
        return res;
    }
    
}