方法一:迭代
我的源代码
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if(!l1)
return l2;
if(!l2)
return l1;
ListNode*head;
if(l1->val>l2->val){
head=l2;
l2=l2->next;
}
else{
head=l1;
l1=l1->next;
}
ListNode*prev=head;
while(1){
if(!l1){
prev->next=l2;
break;
}
if(!l2){
prev->next=l1;
break;
}
if(l1->val>l2->val){
prev->next=l2;
prev=l2;
l2=l2->next;
}
else{
prev->next=l1;
prev=l1;
l1=l1->next;
}
}
return head;
}
};
很臃肿,一如既往,感觉自己提炼问题普适解法的能力很差,边界判断不够一般化。
下面是参考官方题解迭代方法后写出的代码
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode*head=new ListNode;//开辟新空间作为头结点,
不必指向l1或l2,最后返回它的next即可。
ListNode*prev=head;//前驱节点先指向head
while(l1&&l2){//条件判断
if(l1->val>l2->val){
prev->next=l2;
l2=l2->next;
}
else{
prev->next=l1;
l1=l1->next;
}
prev=prev->next;//不论l1还是l2后移,反正prev总是后移了
}
prev->next=l1==nullptr?l2:l1;//这很帅,不像我的很臃肿
return head->next;
}
};
方法二:递归
只能说,妙啊!惊为天人!
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if(!l1)
return l2;
else if(!l2)
return l1;
else if(l1->val<l2->val){
l1->next=mergeTwoLists(l1->next,l2);
return l1;
}
else{
l2->next=mergeTwoLists(l1,l2->next);
return l2;
}
}
};
浪漫