class Solution {
public:
ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) {
ListNode vhead;
if(list1 == nullptr)
{
return list2;
}
if(list2 == nullptr)
{
return list1;
}
ListNode* p,*vp = &vhead ,*l1 = list1,*l2 = list2;
while(l1 != nullptr && l2 != nullptr)
{
if(l1->val < l2->val)
{
p = l1;
l1 = l1->next;
}
else
{
p = l2;
l2 = l2->next;
}
p->next = vp->next;
vp->next = p;
vp = p;
}
p = (l1 == nullptr)?l2:l1;
vp->next = p;
return vhead.next;
}
};
-
函数定义:
cpp复制代码
ListNode* mergeTwoLists(ListNode* list1, ListNode* list2) |
这个函数接受两个参数:两个指向ListNode
的指针,分别代表两个要合并的单链表的头节点。函数返回一个指向合并后链表的头节点的指针。
2.虚拟头节点:
cpp复制代码
ListNode vhead; |
这里创建了一个虚拟头节点vhead
。它的主要目的是简化合并逻辑,避免在合并过程中处理头节点的特殊情况。
3.处理空链表的情况:
cpp复制代码
if(list1 == nullptr) | |
{ | |
return list2; | |
} | |
if(list2 == nullptr) | |
{ | |
return list1; | |
} |
如果其中一个链表为空,则直接返回另一个链表。
4.初始化指针:
cpp复制代码
ListNode* p,*vp = &vhead ,*l1 = list1,*l2 = list2; |
这里初始化了几个指针:
p
:用于临时存储当前要合并的节点。vp
:指向虚拟头节点vhead
。l1
和l2
:分别指向两个要合并的链表的当前节点。
5.合并链表
cpp复制代码
while(l1 != nullptr && l2 != nullptr) | |
{ | |
if(l1->val < l2->val) | |
{ | |
p = l1; | |
l1 = l1->next; | |
} | |
else | |
{ | |
p = l2; | |
l2 = l2->next; | |
} | |
p->next = vp->next; | |
vp->next = p; | |
vp = p; | |
} |
在这个循环中,代码比较l1
和l2
当前节点的值,选择值较小的节点作为下一个合并到结果链表中的节点。然后,更新l1
或l2
以指向下一个节点,并更新vp
以指向结果链表中刚刚添加的节点。
6.处理剩余节点:
cpp复制代码
p = (l1 == nullptr)?l2:l1; | |
vp->next = p; |
循环结束后,如果其中一个链表还有剩余节点,代码将这些剩余节点直接连接到结果链表的末尾。
7.返回结果:
cpp复制代码
return vhead.next; |
最后,函数返回虚拟头节点的next
指针,它指向合并后的链表的第一个实际节点。
这个算法的时间复杂度是O(n + m),其中n和m分别是两个输入链表的长度。这是因为每个节点都只被访问和处理一次。空间复杂度是O(1),因为除了几个指针外,没有使用额外的空间。