LeetCode --- 21. 合并两个有序链表

 

采用归并排序的思路：

两个链表本身都是排序过的，把两个链表从头节点开始，逐个节点开始进行比较，最后剩下的节点接到尾部

 

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
        if(l1 == null) return l2; //一定要初始判断，不然比较大小时会出错
        if(l2 == null) return l1;
        
        ListNode p1 = l1, p2 = l2;
        ListNode head = null, cur = null;//cur指向归并后的当前节点
        
        if(p1.val < p2.val){ //初始化归并后的头节点
            head = p1;
            cur = p1;
            p1 = p1.next;
        }else{
            head = p2;
            cur = p2;
            p2 = p2.next;
        }
        
        while(p1 != null && p2 != null){
            if(p1.val < p2.val){
                cur.next = p1;
                p1 = p1.next;
                cur = cur.next;
            }else{
                cur.next = p2;
                p2 = p2.next;
                cur = cur.next;
            }
        }
        //归并完成，其中一个链表已经归并完，剩下的链表直接加到末尾即可
        if(p1 != null){
            cur.next = p1;
        }
        
        if(p2 != null){
            cur.next = p2;
        }
        return head;
    }
}

 

 

小小优化：

这里也可以设置一个伪头节点，就不需要一开始就判断大小初始化头节点了。

然后直接返回伪头节点的下一个节点：

 

c++:

ListNode *mergeTwoLists(ListNode *l1, ListNode *l2) {
    if (l1 == nullptr) {
        return l2;
    }
    if (l2 == nullptr) {
        return l1;
    }
    ListNode dummy(-1);
    ListNode *p = &dummy;
    for (; l1 && l2; p = p->next) {
        if (l1->val < l2->val) {
            p->next = l1;
            l1 = l1->next;
        } else {
            p->next = l2;
            l2 = l2->next;
        }
    }
    p->next = l1 != nullptr ? l1 : l2;
    return dummy.next;
}