首页 > 编程语言 > 详细

#21 合并排序后的两个链表

时间:2017-06-18 22:35:08      阅读:344      评论:0      收藏:0      [点我收藏+]

思路

使用三个游标:cur指向合并后链表的尾部,l1,l2分别用于遍历两个链表,较小的元素增加到合并后链表。

 

小技巧

使用冗余的头结点可以精简地判断一下情形,其中一个链表,或两个都为空链表。

从而精简代码。

 

朴素代码

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode * head, *ptr;
        
        if (!l1 && !l2) return NULL;
        if (!l1 && l2)  return l2;
        if (l1 && !l2)  return l1;
        
        if (l1->val > l2->val) {
            head = l2;
            l2 = l2->next;
        } else {
            head = l1;
            l1 = l1->next;
        }
        ptr = head;
        while (l1 && l2) {
            if (l1->val <= l2->val) {
                ptr->next = l1;
                l1 = l1->next;
            } else {
                ptr->next= l2;
                l2 = l2->next;
            }
            ptr = ptr->next;
        }
        
        if (l1) {
            ptr->next = l1;
        } else {
            ptr->next = l2;
        }
        
        return head;
    }
};

 

优化代码

class Solution {
public:
    ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
        ListNode dummy(0);
        ListNode *ptr = &dummy;
        while (l1 && l2) {
            if (l1->val <= l2->val) {
                ptr->next = l1;
                l1 = l1->next;
            } else {
                ptr->next= l2;
                l2 = l2->next;
            }
            ptr = ptr->next;
        }
        
        if (l1) {
            ptr->next = l1;
        } else {
            ptr->next = l2;
        }
        
        return dummy.next;
    }
};

 

#21 合并排序后的两个链表

原文:http://www.cnblogs.com/kinsang/p/7045603.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!