题解 | #合并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类
     */

    private ListNode Merge (ListNode pHead1, ListNode pHead2) {
        // write code here
        if (pHead1 == null) {
            return pHead2;
        } else if (pHead2 == null) {
            return pHead1;
        }

        ListNode dummyNode = new ListNode(-1001);  // 用超出范围的哑节点
        ListNode cur = dummyNode;
        ListNode cur1 = pHead1;
        ListNode cur2 = pHead2;

        while (true) {
            if (cur1.val <= cur2.val) {
                cur.next = cur1;
                cur = cur.next;
                cur1 = cur1.next;
            } else {
                cur.next = cur2;
                cur = cur.next;
                cur2 = cur2.next;
            }
            if (cur1 == null) {
                cur.next = cur2;
                break;
            } else if (cur2 == null) {
                cur.next = cur1;
                break;
            }
        }
        return dummyNode.next;
    }

    public ListNode mergeKLists (ArrayList<ListNode> lists) {
        // write code here
        ListNode dummyNode = new ListNode(-1001);
        ListNode pHead1, pHead2;
        pHead1 = dummyNode;
        for(int i=0;i<lists.size();i++){
            pHead2 = lists.get(i);
            pHead1 = Merge(pHead1, pHead2);
        }

        return dummyNode.next;
    }
    }

在合并两个的基础上,循环k次即可;

注意ArrayList类型的索引方法是list.get(i), 获取大小的方法是list.size()

全部评论

相关推荐

我即大橘:耐泡王
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务