这道题涉及一个很基本的操作——合并,合并在归并算法中很常用!
题目:将两个有序链表合并为一个有序的链表,新的有序链表使用之前的两个链表拼接而成。
这个题的思路很简单,让我们直接看代码,同时要注意代码的鲁棒性!
要注意l1或者l2为nullptr的情况,不要让程序崩溃!
/**
* 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)
{
//我们最终要返回这个head的next
ListNode head = ListNode(0);
ListNode *curr = &head;
while(l1 != nullptr || l2 != nullptr)
{
//处理l1或者l2指向nullptr的情况
if(!l1) {curr->next = l2, l2 = l2->next;}
else if(!l2) {curr->next = l1, l1 = l1->next;}
//如果l1和l2都指向具体的节点,那么就比较它们的大小来决定curr的下一个指向谁
else
{
l1->val < l2->val ?
(curr->next = l1, l1 = l1->next) : (curr->next = l2, l2 = l2->next);
}
curr = curr->next;
}
return head.next;
}
};