第十三周题解

23. 合并K个排序链表

class Solution {
    //因为给的链表都是排序链表，所以当前节点一定比下一个节点小，每一次从最小堆中出堆一个结点，只要这个结点不是尾结点，都将这个结点的下一个结点继续入堆
   public ListNode mergeKLists(ListNode[] lists) {
        if (lists.length == 0)
        	return null;
        //最小堆，堆中的数都按从小到大排列
        PriorityQueue<ListNode> queue = new PriorityQueue<>(new Comparator<ListNode>() {
            @Override
            public int compare(ListNode o1, ListNode o2) {
                return o1.val - o2.val;
            }
        });
        //创建一个链表的头结点
        ListNode head = new ListNode(0);
        ListNode p = head;
        //将给定的k个链表的头结点入最小堆
        for (ListNode node : lists) {
            if (node != null) queue.add(node);
        }
        while (!queue.isEmpty()) {
            //获取并移除此队列的头，即给定的k个排序链表中头结点最小的数
            p.next = queue.poll();
            p = p.next;
            //如果这个结点不是尾结点，将这个结点的下一个结点入最小堆
            if (p.next != null)
                queue.add(p.next);
        }
        return head.next;
    }
}

321.拼接最大数

class Solution {
    public int[] maxNumber(int[] nums1, int[] nums2, int k) {
        int m = nums1.length, n = nums2.length;
        int[] res = new int[k];
        //先从nums2中取k个数，若不够，则取n个数，从nums1中取k-n个数，然后依次增减
        for (int i = Math.max(0, k - n); i <= k && i <= m; i++) {
            //将两个数组组合起来，使其数值最大
            int[] arr = merge(maxArr(nums1, i), maxArr(nums2, k - i), k);
            //比较每种组合的大小，取大的
            if (gt(arr, 0, res, 0)) res = arr;
        }
        return res;
    }

    //从给定数组中按顺序取得k个最大的数，返回值为取得的数组成的数组
    private int[] maxArr(int[] nums, int k) {
        int n = nums.length;
        int[] res = new int[k];
        //先将nums[0]加入的最大组合中，遍历nums数组，依次判断是否加入最大数组，或是否要替换掉已加入最大数组中的数
        //j为当前数组中下标i之前有多少个数加入到最大组合中
        for (int i = 0, j = 0; i < n; i++) {
            //n-i为nums中还剩几个数没有判断，n-i+j为当前最多有几个数，如果下一个数比已选取的数大的话，则将其替换
            while (n - i + j > k && j > 0 && nums[i] > res[j-1])
                    j --;
            if (j < k)
                res[j ++] = nums[i];
        }
        return res;
    }
    //拼接两个数组，得到结果最大的数组
    private int[] merge(int[] nums1, int[] nums2, int k) {
        int[] res = new int[k];
        for (int i = 0, j = 0, r = 0; r < k; r ++){
            //判断两个数组当前位置的数的大小，取大的，已取得的数组下标+1
            res[r] = gt(nums1, i, nums2, j) ? nums1[i++] : nums2[j++];
        }
        return res;
    }

    //比较两数组相应位置大小,相等就一直跳过,直到不相等就比较
    private boolean gt(int[] nums1, int i, int[] nums2, int j) {
        while (i < nums1.length && j < nums2.length && nums1[i] == nums2[j]) {
            i++;
            j++;
        }
        //如果两数组前面的数一样，则先从nums1中取
        return j == nums2.length || (i < nums1.length && nums1[i] > nums2[j]);
    }
}