题解 | #合并k个已排序的链表# 归并排序

合并k个已排序的链表

https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 *   public ListNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param lists ListNode类ArrayList 
     * @return ListNode类
     */
    public ListNode mergeKLists (ArrayList<ListNode> lists) {
        if (lists.isEmpty()) {
            return null;
        }
        if (lists.size() == 1) {
            return lists.get(0);
        }
        int size = lists.size();
        int leftSize = size / 2;
        int rightSize = size - leftSize;
        ArrayList<ListNode> leftPart = new ArrayList<>();
        ArrayList<ListNode> rightPart = new ArrayList<>();
        for (int i = 0; i < leftSize; i++) {
            leftPart.add(lists.get(i));
        }
        for (int i = leftSize; i < size; i++) {
            rightPart.add(lists.get(i));
        }
        ListNode leftRes = mergeKLists(leftPart);
        ListNode rightRes = mergeKLists(rightPart);
        ListNode hair = new ListNode(0);
        ListNode pointer = hair;
        while (leftRes != null || rightRes != null) {
            if (leftRes == null) {
                pointer.next = rightRes;
                return hair.next;
            } else if (rightRes == null) {
                pointer.next = leftRes;
                return hair.next;
            } else if (leftRes.val < rightRes.val) {
                pointer.next = leftRes;
                leftRes = leftRes.next;
            } else {
                pointer.next = rightRes;
                rightRes = rightRes.next;
            }
            pointer = pointer.next;
        }
        return hair.next;
    }
}

全部评论

相关推荐

点赞 评论 收藏
分享
双非一本失业第二年:《机器视觉垃圾分类》
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务