LeetCode 题目连接:https://leetcode-cn.com/problems/he-bing-liang-ge-pai-xu-de-lian-biao-lcof/
题目描述:
输入两个递增排序的链表,合并这两个链表并使新链表中的节点仍然是递增排序的。
示例1:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
限制:
0 <= 链表长度 <= 1000
思路:
很简单的题,直接定义一个头,然后两个指针相互比较即可。
问题
再最初版本出现了问题,最开始的想法很简单,定义一个头pre
用他去遍历两个链表,再用一个ans
来记录pre
的头部,然后问题出现了,使用pre=pre->next, pre=l2
时,发现ans
并不能完全表达pre
的值
ListNode *pre = NULL;
if(l1->val <= l2->val) {
pre = l1;
l1 = l1->next;
}else {
pre = l2;
l2 = l2->next;
}
ListNode* ans = pre;
pre = pre->next;
while(l1 != NULL && l2 != NULL) {
if(l1->val <= l2->val) {
pre = l1;
l1 = l1->next;
}else {
pre = l2;
l2 = l2->next;
}
pre = pre->next;
}
类似如此,ans
返回的时第一次的pre
值
为什么
第一次ans=pre
,此时ans
与pre
指向同一个地址,类似如此
当执行pre=pre->next
时
当这时执行到pre=l2
时,
会发现pre
指针直接指向l2
,而不会带着next指针一起指向l2
,这就造成了pre
自己走完了一整个链表而ans
却一直指向开始的位置。
解决方案
指的时候带上next
就可以了
ListNode* pre = new ListNode(0);
ListNode* cur = pre;
while(l1 != NULL && l2 != NULL) {
if(l1->val <= l2->val) {
cur->next = l1;
l1 = l1->next;
}else {
cur->next = l2;
l2 = l2->next;
}
cur = cur->next;
}
AC代码
/**
* 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) {
ListNode* pre = new ListNode(0);
ListNode* cur = pre;
while(l1 != NULL && l2 != NULL) {
if(l1->val <= l2->val) {
cur->next = l1;
l1 = l1->next;
}else {
cur->next = l2;
l2 = l2->next;
}
cur = cur->next;
}
if(l1 != NULL) cur->next = l1;
if(l2 != NULL) cur->next = l2;
return pre->next;
}
};