力扣21. 合并两个有序链表(链表,迭代)
https://leetcode-cn.com/problems/merge-two-sorted-lists/
将两个升序链表合并为一个新的升序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4
方法 :迭代
有点像归排中合并的算法
动画:
我们可以用迭代的方法来实现上述算法。我们假设 l1 元素严格比 l2元素少,我们可以将 l2 中的元素逐一插入 l1 中正确的位置。
算法:
首先,我们设定一个哨兵节点 "prehead" ,这可以在最后让我们比较容易地返回合并后的链表。我们维护一个 prev 指针,我们需要做的是调整它的 next 指针。然后,我们重复以下过程,直到 l1 或者 l2 指向了 null :如果 l1 当前位置的值小于等于 l2 ,我们就把 l1 的值接在 prev 节点的后面同时将 l1 指针往后移一个。否则,我们对 l2 做同样的操作。不管我们将哪一个元素接在了后面,我们都把 prev 向后移一个元素。
在循环终止的时候, l1 和 l2 至多有一个是非空的。由于输入的两个链表都是有序的,所以不管哪个链表是非空的,它包含的所有元素都比前面已经合并链表中的所有元素都要大。这意味着我们只需要简单地将非空链表接在合并链表的后面,并返回合并链表。
时间复杂度:O(n+m)。 因为每次调用递归都会去掉 l1 或者 l2 的头元素(直到至少有一个链表为空),函数 mergeTwoList 中只会遍历每个元素一次。所以,时间复杂度与合并后的链表长度为线性关系。
空间复杂度:O(1) 。迭代的过程只会产生几个指针,所以它所需要的空间是常数级别的。
#include "stdafx.h"
struct ListNode
{
int val;
ListNode *next;
ListNode(int x) : val(x), next(NULL) {}
};
class Solution
{
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2)
{
ListNode* head = new ListNode(0);
ListNode* cur = head;
while (l1!=NULL && l2 != NULL)
{
if ((l1->val)<=(l2->val))
{
cur->next = l1;
cur = cur->next;
l1 = l1->next;
}
else
{
cur->next = l2;
cur = cur->next;
l2 = l2->next;
}
}
while (l1 != NULL)
{
cur->next = l1;
cur = cur->next;
l1 = l1->next;
}
while (l2 != NULL)
{
cur->next = l2;
cur = cur->next;
l2 = l2->next;
}
return head->next;
}
};
int main()
{
Solution s;
ListNode l1[3] = { 1,2,4 };
l1[0].next = &l1[1]; l1[1].next = &l1[2];
ListNode l2[3] = { 1,3,4 };
l2[0].next = &l2[1]; l2[1].next = &l2[2];
auto result = s.mergeTwoLists(l1, l2);
return 0;
}