题目

Merge Two Sorted Lists

思路

主要考虑一些特殊情况,例如有一个链表为空,或者长度不同。这里的方法是创建了一个新的链表头节点,通过遍历原来的两个链表来选择较小的插入到新链表。当一个链表提前结束时,将next指向下一个链表全部插入。最后返回新列表头节点的下一个节点。

代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        if (!l1) return l2; // list 1 is empty
        if (!l2) return l1; // list 2 is empty
        ListNode *dummy = new ListNode(0);
        ListNode *currentTail = dummy; // current tail of the new list
        while (l1 && l2){
            if (l1->val <= l2->val){
                currentTail->next = l1;
                currentTail = l1;
                if (l1->next)
                    l1 = l1->next;
                else{
                    l1->next = l2;
                    return dummy->next;
                }
            }
            else {
                currentTail->next = l2;
                currentTail = l2;
                if (l2->next)
                    l2 = l2->next;
                else{
                    l2->next = l1;
                    return dummy->next;
                }
            }
        }
    }
};