题目
思路
主要考虑一些特殊情况,例如有一个链表为空,或者长度不同。这里的方法是创建了一个新的链表头节点,通过遍历原来的两个链表来选择较小的插入到新链表。当一个链表提前结束时,将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;
}
}
}
}
};