004合并两个有序链表

最新推荐文章于 2024-07-18 17:57:16 发布

zk_lar

最新推荐文章于 2024-07-18 17:57:16 发布

阅读量66

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/zk_lar/article/details/125746830

版权

链表数据结构

算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题目描述

在这里插入图片描述

代码实现

不带头结点方式

/*
合并两个有序链表
思路:依次比较两个链表中的节点，每次取较小的节点。尾插到新链表即可。
	因为尾插每次都需要找尾，所以可以定义一个tail来记录尾部
*/

struct ListNode
{
	int val;
	struct ListNode* next;
};

struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2)
{
	// 判断list1和list2是否为空，其中一个为空，返回另一个
	if (!list1)
		return list2;
	if (!list2)
		return list1;

	struct ListNode *head = NULL, *tail = NULL;
	// 当head为空时，先取一个节点作为第一个
	if (list1->val < list2->val)
	{
		head = tail = list1;
		list1 = list1->next;
	}
	else
	{
		head = tail = list2;
		list2 = list2->next;
	}

	while (list1 && list2)
	{
		if (list1->val < list2->val)
		{
			tail->next = list1;
			tail = list1;
			list1 = list1->next;
		}
		else
		{
			tail->next = list2;
			tail = list2;
			list2 = list2->next;
		}
	}

	// 其中一个链表先为空，将另一个链表剩余的全部节点尾插到新链表
	if (list1)
	{
		tail->next = list1;
	}
	if (list2)
	{
		tail->next = list2;
	}

	return head;
}

带头结点方式

/*
合并两个有序链表
思路:依次比较两个链表中的节点，每次取较小的节点。尾插到新链表即可。
	因为尾插每次都需要找尾，所以可以定义一个tail来记录尾部
*/

struct ListNode
{
	int val;
	struct ListNode* next;
};

struct ListNode* mergeTwoLists(struct ListNode* list1, struct ListNode* list2)
{
	// 判断list1和list2是否为空，其中一个为空，返回另一个
	if (!list1)
		return list2;
	if (!list2)
		return list1;

	struct ListNode *head = NULL, *tail = NULL;
	// 开辟一个头节点
	head = tail = (struct ListNode*)malloc(sizeof(struct ListNode));

	while (list1 && list2)
	{
		if (list1->val < list2->val)
		{
			tail->next = list1;
			tail = list1;
			list1 = list1->next;
		}
		else
		{
			tail->next = list2;
			tail = list2;
			list2 = list2->next;
		}
	}

	// 其中一个链表先为空，将另一个链表剩余的全部节点尾插到新链表
	if (list1)
	{
		tail->next = list1;
	}
	if (list2)
	{
		tail->next = list2;
	}

	struct ListNode* list = head->next;
	// 释放头节点
	free(head);
	return list;
}