21. 合并两个有序链表

题目来源:力扣

题目描述:将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。 

示例 1:

输入:l1 = [1,2,4], l2 = [1,3,4]
输出:[1,1,2,3,4,4]

示例 2:

输入:l1 = [], l2 = []
输出:[]

示例 3:

输入:l1 = [], l2 = [0]
输出:[0]

 思路:取两个链表里小的节点插入新的链表,比如示例1的1,2,4和1,3,4,取1插入新节点,取1插入新节点,取2插入新节点,以此类推,直到一个链表全部插完,剩下的那个链表直接链接在新链表的尾部即可。

struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
    if(list1==NULL){
        return list2;
    }
    if(list2==NULL){
        return list1;
    }
    struct ListNode* head=NULL;
    struct ListNode* tail=NULL;
    while(list1&&list2){
        if(list1->val<list2->val){
            if(tail==NULL){
                head=tail=list1;
            }else{
                tail->next=list1;
                tail=tail->next;
            }
            list1=list1->next;
        }else{
            if(tail==NULL){
                head=tail=list2;
            }else{
                tail->next=list2;
                tail=tail->next;
            }
            list2=list2->next;
        }
    }
    if(list1){
        tail->next=list1;
    }
    if(list2){
        tail->next=list2;
    }
    return head;
}

 注意:要判断返回空链表。循环控制条件为&&而不是||。插入第一个节点时要判断,因为新链表是空的。

运行结果:

 由于代码过于冗杂,我们对其进行优化

优化一:

struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
    if(list1==NULL){
        return list2;
    }
    if(list2==NULL){
        return list1;
    }
    struct ListNode* head=NULL;
    struct ListNode* tail=NULL;
    if(list1->val<list2->val){
        head=tail=list1;
        list1=list1->next;
    }else{
        head=tail=list2;
        list2=list2->next;
    }
    while(list1&&list2){
        if(list1->val<list2->val){
            tail->next=list1;                     
            list1=list1->next;
        }else{           
            tail->next=list2;
            list2=list2->next;
        }
        tail=tail->next;         
    }
    if(list1){
        tail->next=list1;
    }
    if(list2){
        tail->next=list2;
    }
    return head;
}

我们在外部就解决新链表的头节点问题,同时提取循环内部公共语句tail=tail->next放到循环底部。

优化二:

struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2){
    if(list1==NULL){
        return list2;
    }
    if(list2==NULL){
        return list1;
    }
    struct ListNode* head=NULL;
    struct ListNode* tail=NULL;
    head=tail=(struct ListNode*)malloc(sizeof(struct ListNode));
    while(list1&&list2){
        if(list1->val<list2->val){
            tail->next=list1;                     
            list1=list1->next;
        }else{           
            tail->next=list2;
            list2=list2->next;
        }
        tail=tail->next;         
    }
    if(list1){
        tail->next=list1;
    }
    if(list2){
        tail->next=list2;
    }
    struct ListNode* first=head->next;
    free(head);
    return first;
}

我们给head和tail最开始初始化一个哨兵位的头节点,之后只要进行尾插即可,最后我们释放这个头节点,返回头节点的next

运行结果:

 

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值