题目要求截图:
已经想到了就是只利用两个指针比较大小然后换next。但是其中特殊的情况没有考虑到位。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
if(list1 == nullptr && list2 == nullptr) return nullptr;
if(list1 == nullptr) return list2;
if(list2 == nullptr) return list1;
ListNode* res = list1->val <= list2->val? list1:list2;
while(list1 != nullptr && list2 != nullptr){
if(list1->val <= list2->val){
ListNode* temp = list1->next;
list1->next = list2;
list1 = temp;
}else{
ListNode* temp = list2->next;
list2->next = list1;
list2 = temp;
}
}
return res;
}
};
比如例子[5], 和[1,2,3] 就会输出 [1,5]
主要思想还是要再加一个pre,更新list1和list2只是用来比较大小的,pre是真正用来变换指针指来指去的中间变量。之前的做法里面没法需要比较list1比list2小,还得看list1->next是不是还是比list2小。这种办法还需要调用next,在while循环判断的地方容易造成越界,不好写while的判断条件。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
if(list1 == nullptr && list2 == nullptr) return nullptr;
if(list1 == nullptr) return list2;
if(list2 == nullptr) return list1;
ListNode* res = list1->val <= list2->val? list1:list2;
ListNode* pre = new ListNode(-1);
pre->next = res;
while(list1 != nullptr && list2 != nullptr){
if(list1->val > list2->val){
pre->next = list2;
list2 = list2->next;
}else{
pre->next = list1;
list1 = list1->next;
}
pre = pre->next;
}
if(list1 == nullptr) pre->next = list2;
if(list2 == nullptr) pre->next = list1;
return res;
}
};
既然判断后面不行那就在前面多加一个头,让逻辑多往后跑一位(list1和list2),实际的操作就在现在的pre里面做,这样就能避免向后检查。但是最后一定有一个便利到了nullptr,只需要把最后连上即可。
链表第一次做,一道连接链表的题