我写的复杂了,主要原因是没有意识到一个问题,就是链表的链接只需要接一个就行了,而不需要一个节点一个节点的去接。
题解有两种方法,迭代和递归
#include<iostream>
using namespace std;
struct ListNode {
int val;
ListNode *next;
ListNode() : val(0), next(nullptr) {}
ListNode(int x) : val(x), next(nullptr) {}
ListNode(int x, ListNode *next) : val(x), next(next) {}
};
class Solution {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if (l1 == nullptr && l2 == nullptr) {
return nullptr;
}
ListNode* head=new ListNode();
ListNode* ans = head;
while (l1 != nullptr || l2 != nullptr) {
if (l1 == nullptr) {
ans->val = l2->val;
if (l2->next != nullptr) {
ans->next = new ListNode();
ans = ans->next;
}
l2 = l2->next;
}
else if (l2 == nullptr) {
ans->val = l1->val;
if (l1->next != nullptr) {
ans->next = new ListNode();
ans = ans->next;
}
l1 = l1->next;
}//其实这些都很多余
else {
if (l1->val < l2->val) {
ans->val = l1->val;
if (l1->next != nullptr||l2!=nullptr) {
ans->next = new ListNode();
ans = ans->next;
}
l1 = l1->next;
}
else
{
ans->val = l2->val;
if (l1!=nullptr||l2->next != nullptr) {
ans->next = new ListNode();
ans = ans->next;
}
l2 = l2->next;
}
}
}
return head;
}
};
//递归方法
class Solution1 {
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
if (l1 == nullptr) {
return l2;
}
else if (l2 == nullptr) {
return l1;
}//递归终点
else if (l1->val < l2->val) {
l1->next = mergeTwoLists(l1->next, l2);
return l1;
}
else {
l2->next = mergeTwoLists(l1, l2->next);
return l2;
}
}
};
//迭代方法,题解方法 比我写的简单很多
class Solution2{
public:
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) {
ListNode* preHead = new ListNode(-1);
ListNode* prev = preHead;//这个处理方法和我一样,因为要返回头节点,所以新建一个
while (l1 != nullptr && l2 != nullptr) {
if (l1->val < l2->val) {
prev->next = l1;
l1 = l1->next;
}
else {
prev->next = l2;
l2 = l2->next;
}
prev = prev->next;
}
// 合并后 l1 和 l2 最多只有一个还未被合并完,我们直接将链表末尾指向未合并完的链表即可
prev->next = l1 == nullptr ? l2 : l1;//如果有没合并完的,接上去
return preHead->next;//从第二个开始接的
}
};
void print(ListNode* head)//输出结果链表
{
ListNode* p;
p = head;
while (p != NULL)
{
cout << p->val << " ";
p = p->next;
}
}
int main() {
Solution test;
ListNode* l1 = new ListNode(1);
l1->next = new ListNode(2);
l1->next->next = new ListNode(4);
ListNode* l2 = new ListNode(1);
l2->next = new ListNode(3);
l2->next->next = new ListNode(4);
ListNode* l3 = nullptr;
ListNode* l4 = nullptr;
ListNode *ans=test.mergeTwoLists(l3, l4);
print(ans);
}