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

合并k个已排序的链表

https://www.nowcoder.com/practice/65cfde9e5b9b4cf2b6bafa5f3ef33fa6?tpId=295&tags=&title=&difficulty=0&judgeStatus=0&rp=0&sourceUrl=%2Fexam%2Foj%3Ftab%3D%25E7%25AE%2597%25E6%25B3%2595%25E7%25AF%2587%26topicId%3D295

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 mergeTwo(ListNode pHead1, ListNode pHead2){
        ListNode head = new ListNode(-1);
        ListNode cur = head;
        while(pHead1 != null && pHead2 != null){
            if(pHead1.val <= pHead2.val){
                cur.next = pHead1;
                pHead1 = pHead1.next;
            }else{
                cur.next = pHead2;
                pHead2 =  pHead2.next;
            }
            cur = cur.next;
        }
        if(pHead1 != null){
            cur.next = pHead1;
        }else{
            cur.next = pHead2;
        }
        return head.next;
    }

    public ListNode merge(ArrayList<ListNode> lists, int left, int right){
        if(left > right){
            return null;
        }
        if(left == right){ 
            return lists.get(left);
        }
        int mid = (left + right)/2;
        //对半划分,将划分后的子问题合并成新的链表
        return mergeTwo(merge(lists,left,mid), merge(lists,mid + 1,right));
    }

    public ListNode mergeKLists (ArrayList<ListNode> lists) {
        // write code here
        return merge(lists,0,lists.size()-1);
    }
}

全部评论

相关推荐

拉丁是我干掉的:把上海理工大学改成北京理工大学。成功率增加200%
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
11-21 17:16
科大讯飞 算法工程师 28.0k*14.0, 百分之三十是绩效,惯例只发0.9
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务