合并排序链表+指针指向问题

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的值

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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,此时anspre指向同一个地址,类似如此

img1

当执行pre=pre->next

当这时执行到pre=l2时,

会发现pre指针直接指向l2,而不会带着next指针一起指向l2,这就造成了pre自己走完了一整个链表而ans却一直指向开始的位置。

解决方案

指的时候带上next就可以了

1
2
3
4
5
6
7
8
9
10
11
12
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代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
/**
* 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;
}
};
---------Thanks for your attention---------