题目
将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成
的
一、思路详解
第一步
比较两个链表的每一位数,小的就放前面
核心思路就是这样一次一次的比较,然后放值
由此可以看出我们需要循环遍历链表来比较
//只是写出交换的过程并未循环交换
if(list1->val > list2->val)
{
tail->next = list2;
tail = list2;
list2 = list2->next;
}
else
{
tail->next = list1;
tail = list1;
list1 = list1->next;
}
第二步
思考代码什么时候停止
从图片我们可以看出单一个链表被遍历结束时循环条件就结束了,当循环之后将另一个还未结束的链表链接在tail后面
因为刚开始创建的head(头节点为空)所以我们
要在赋值的时候判断一下避免野指针问题。
struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2)
{
struct ListNode* head = NULL;
struct ListNode* tail = NULL;
while(list1 && list2)
{
if(list1->val > list2->val)
{
if(head == NULL)
{
head = tail = list2;
}
else
{
tail->next = list2;
tail = list2;
}
list2 = list2->next;
}
else
{
if(head == NULL)
{
head = tail = list1;
}
else
{
tail->next = list1;
tail = list1;
}
list1 = list1->next;
}
}
if(list1)
{
tail->next = list1;
tail = list1;
list1 = list1->next;
}
if(list2)
{
tail->next = list2;
tail = list2;
list2 = list2->next;
}
return head;
}
第三步
思考代码是否是所有情况都适合
我们会发现当其中一个链表为空,或者链表全部为空时会出错
我们假设list1为空此时while循环便会被跳过,最后返回的head会是一个空指针,此时代码就会出错。
所以一开始就要判断两个链表是否为空的情况
二、接口形OJ完整代码
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* struct ListNode *next;
* };
*/
struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2)
{
if(list1 == NULL)
{
return list2;
}
if(list2 == NULL)
{
return list1;
}
struct ListNode* head = NULL;
struct ListNode* tail = NULL;
while(list1 && list2)
{
if(list1->val > list2->val)
{
if(head == NULL)
{
head = tail = list2;
}
else
{
tail->next = list2;
tail = list2;
}
list2 = list2->next;
}
else
{
if(head == NULL)
{
head = tail = list1;
}
else
{
tail->next = list1;
tail = list1;
}
list1 = list1->next;
}
}
if(list1)
{
tail->next = list1;
tail = list1;
list1 = list1->next;
}
if(list2)
{
tail->next = list2;
tail = list2;
list2 = list2->next;
}
return head;
}
力扣链接:https://leetcode-cn.com/problems/merge-two-sorted-lists/description/