算法

合并两个有序链表

2020-10-17  本文已影响0人  小鱼嘻嘻

问题1

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

原理

代码

class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        if(l1==null){
            return l2;
        }
        if(l2==null){
            return l1;
        }
        ListNode newHead = new ListNode(-1);
        ListNode run = newHead;

        while(l1!=null&&l2!=null){
            if(l1.val<l2.val){
                run.next =l1;
                l1 = l1.next;
            }else{
                run.next = l2;
                l2 = l2.next;
            }
            run = run.next;
        }
        if(l1!=null){
            run.next = l1;
        }
        if(l2!=null){
            run.next = l2;
        }
        return newHead.next;
    }
}

注意事项

问题2

给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。

原理

代码

class Solution {
    public ListNode mergeKLists(ListNode[] lists) {
        if(lists==null||lists.length<=0){
            return null;
        }

        PriorityQueue<ListNode> pq = new PriorityQueue<>((o1,o2)->o1.val-o2.val);
        for(ListNode l:lists){
            if(l!=null){
                pq.add(l);
            }
        }

        ListNode newHead = new ListNode(-1);
        ListNode run = newHead;

        while(!pq.isEmpty()){
            ListNode cur = pq.poll();
            run.next = cur;
            run = run.next;
            if(cur.next!=null){
                pq.add(cur.next);
            }
        }
        return newHead.next;
    }
}

注意事项

上一篇 下一篇

猜你喜欢

热点阅读