合并k个已排序的链表

心文花雨

于 2022-06-13 09:59:20 发布

阅读量165

点赞数

分类专栏： golang 面试题文章标签：链表数据结构算法

本文链接：https://blog.youkuaiyun.com/xinwenhuayu/article/details/125254635

版权

面试题同时被 2 个专栏收录

46 篇文章

订阅专栏

golang

9 篇文章

订阅专栏

这篇博客介绍了如何合并多个升序排列的链表，并返回合并后的新链表。博主给出了一个Go语言的代码实现，通过递归地合并两个链表，最终将所有链表合并成一个。主要涉及数据结构和算法的知识，特别是链表操作和排序算法的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：合并k个已排序的链表_牛客题霸_牛客网

描述

合并 k 个升序的链表并将结果作为一个升序的链表返回其头节点。

示例

输入：

    [{1,2,3},{4,5,6,7}]

返回值：

    {1,2,3,4,5,6,7}

Go代码实现

func merge2(pHead1 *ListNode, pHead2 *ListNode) *ListNode {
	result := &ListNode{}
	cur := result
	for pHead1 != nil && pHead2 != nil {
		if pHead1.Val >= pHead2.Val {
			cur.Next = pHead2
			pHead2 = pHead2.Next
		} else {
			cur.Next = pHead1
			pHead1 = pHead1.Next
		}
		cur = cur.Next
	}

	if pHead1 == nil {
		cur.Next = pHead2
	} else if pHead2 == nil {
		cur.Next = pHead1
	}

	return result.Next
}

func mergeKLists(lists []*ListNode) *ListNode {
	// write code here
	if len(lists) < 1 {
		return nil
	}

	for len(lists) > 1 {
		for i := 0; i < len(lists)-1; i++ {
			lists[i] = merge2(lists[i], lists[len(lists)-1])
			lists[len(lists)-1] = nil
			lists = lists[:len(lists)-1]
		}
	}

	return lists[0]
}