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


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,此时anspre指向同一个地址,类似如此

img1

当执行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;
    }
};

文章作者: Mug-9
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 Mug-9 !
评论
  目录