您的位置:首页 > 其它

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

2017-06-18 22:08 281 查看
思路

使用三个游标: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;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: