牛客-c语言刷题-合并两个排序的链表

题目连接：合并两个排序的链表_牛客题霸_牛客网 (nowcoder.com)

题目描述：

输入两个递增的链表，单个链表的长度为n，合并这两个链表并使新链表中的节点仍然是递增排序的。

数据范围： 0≤n≤1000，−1000≤节点值≤1000
要求：空间复杂度 O(1)，时间复杂度 O(n)

如输入{1,3,5},{2,4,6}时，合并后的链表为{1,2,3,4,5,6}，所以对应的输出为{1,2,3,4,5,6}，转换过程如下图所示：

或输入{-1,2,4},{1,3,4}时，合并后的链表为{-1,1,2,3,4,4}，所以对应的输出为{-1,1,2,3,4,4}，转换过程如下图所示：

题目解析：

#include "stdio.h"
#include "stdlib.h"


/*
 *BM4 合并两个排序的链表

输入两个递增的链表，单个链表的长度为n，合并这两个链表并使新链表中的节点仍然是递增排序的。
数据范围： 0≤n≤1000，−1000≤节点值≤1000
要求：空间复杂度 O(1)O(1)，时间复杂度 O(n)O(n)

如输入{1,3,5},{2,4,6}时，合并后的链表为{1,2,3,4,5,6}，所以对应的输出为{1,2,3,4,5,6}，
*/

/**************************code*****************************************/
 struct ListNode {
	int val;
	struct ListNode *next;
 };

struct ListNode* Merge(struct ListNode* pHead1, struct ListNode* pHead2 ) {
    // write code here
    struct ListNode *mainList, *cur, *tmp;  //mainList为 第一个元素较小的链表
    struct ListNode *insertList;            //insertList为 第一个元素较大的链表
    if(!pHead1 && pHead2)   return pHead2;  //特殊输入处理
    if(!pHead2 && pHead1)   return pHead1;
    if(!pHead2 && !pHead1)  return NULL;
    if(pHead1->val <= pHead2->val){         //判断一个元素的大小，将pHead1\2赋值给mainlist insertlist
        mainList = pHead1, insertList = pHead2;
    }else{
        mainList = pHead2, insertList = pHead1;
    }
    cur = mainList;                         //当前指针

    while(cur->next || insertList){         //只要当前元素和插入链表 有任一是存在的，则进行循环
        if(!cur->next && insertList){       //mainlist的已经到末尾了，则直接插入insertList的值并退出
            cur->next = insertList;         //这里不需要考虑insertList的值是否大于cur，因为由前面赋值mainList insertList的
            break;                          //  逻辑可知insertList的值必小于cur
        }else if(!insertList){              //当insertList不存在，则退出
            break;
        }else if(cur->next && insertList && cur->next->val > insertList->val){
            tmp = cur->next;                //当cur的下一个值存在且insertList存在，则判断insert的值是否小于cur->next的值
            cur->next = insertList;         // 小于则进行插入
            insertList = insertList->next;
            cur = cur->next;                //cur指向插入的值
            cur->next = tmp;                //将插入的值next指向原来cur的next
        }else{
            cur = cur->next;                //当cur的下一个值存在且insertList存在，且insert的值大于cur->next的值
        }                                   //则cur移到下一个
    }
    return mainList;
}

/**************************end******************************************/

int main ()
{
    struct ListNode LN13={5,NULL};
    struct ListNode LN12={3,&LN13};
    struct ListNode LN1 ={1,&LN12};

    struct ListNode LN23={6,NULL};
    struct ListNode LN22={4,&LN23};
    struct ListNode LN2 ={2,&LN22};
    
    struct ListNode LN33={3,NULL};
    struct ListNode LN32={2,&LN33};
    struct ListNode LN3 ={1,&LN32};

    struct ListNode LN43={3,NULL};
    struct ListNode LN42={3,&LN43};
    struct ListNode LN4 ={3,&LN42};

    struct ListNode *pLN = Merge(&LN3, &LN4);

    return 0;
}