码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2017-06-18 22:44:57      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:优化   while   ptr   null   sts   merge   精简   ret   solution   

思路

使用三个游标: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 合并排序后的两个链表

标签:优化   while   ptr   null   sts   merge   精简   ret   solution   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!