剑指offer-合并两个排序的链表

最新推荐文章于 2023-07-02 17:54:14 发布

原创最新推荐文章于 2023-07-02 17:54:14 发布 · 297 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

62 篇文章

订阅专栏

本文介绍了一种就地合并两个已排序链表的方法，通过调整指针实现空间复杂度为O(1)的合并操作。讨论了传统merge思路及其实现，并针对特定情况进行详细解释。

问题

题目：[合并两个排序的链表]

思路

传统merge的思路不说了，因为此时空间的复杂度是O(N).
主要说就地合并的思路。当然，这个题其实牛客的测试用例有问题，考虑不全面。
基本思想就是当p的当前值比q的当前值小的时候，q的上一个元素要指向p。当然，这有个前提是q的上一个元素不能指向pre_p，否则没有没有指向的意义。比如 1->3->5, 2->100->101。牛客没有考虑这种测试用例

代码

/*
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};*/
class Solution {
public:
    ListNode* Merge(ListNode* pHead1, ListNode* pHead2)
    {
        ListNode* p = pHead1;
        ListNode* q = pHead2;

        if(!p && !q) return NULL;
        if(p && !q) return p;
        if(!p && q) return q;


        ListNode* p_pre = NULL;
        ListNode* q_pre = NULL;
        while(p&&q){
            if( p->val < q->val ){
                if(q_pre && q_pre->next != p_pre ){
                    q_pre->next = p;
                }
                p_pre = p;
                p = p->next;
            }
            else{
                if(p_pre && p_pre->next != q_pre ){
                    p_pre->next = q;
                }
                q_pre = q;
                q = q->next;
            }
        }
        if(p) q_pre->next = p;
        if(q) p_pre->next = q;

        if(pHead1->next == pHead2) return pHead1;
        else return pHead2;

    }
};