合并k个有序链表
链接:题目链接
题目描述
给你一个链表数组,每个链表都已经按升序排列。
请你将所有链表合并到一个升序链表中,返回合并后的链表。
示例 1:
输入:lists = [[1,4,5],[1,3,4],[2,6]]
输出:[1,1,2,3,4,4,5,6]
解释:链表数组如下:
[
1->4->5,
1->3->4,
2->6
]
将它们合并到一个有序链表中得到。
1->1->2->3->4->4->5->6
解题思路—小根堆,分治
这题和合并两个有序链表很像,但这里是k个,我们只需要进行分治思想应用在这里即可,每次只处理2个链表
代码实现
class Solution:
def mergeKLists(self, lists: List[ListNode]) -> ListNode:
if not lists or len(lists) == 0:
return None
import heapq
heap = []
# 首先 for 嵌套 while 就是将所有元素都取出放入堆中
for node in lists:
while node:
heapq.heappush(heap, node.val)
node = node.next
dummy = ListNode(None)
cur = dummy
# python的堆实现的是最小堆,也就是小根堆
# 依次将堆中的元素取出(因为是小顶堆,所以每次出来的都是目前堆中值最小的元素),然后重新构建一个列表返回
while heap:
temp_node = ListNode(heappop(heap))
cur.next = temp_node
cur = temp_node
return dummy.next
# 优先队列就是堆
class Solution:
def merge(self, node_a, node_b):
dummy = ListNode(None)
cursor_a, cursor_b, cursor_res = node_a, node_b, dummy
while cursor_a and cursor_b: # 对两个节点的 val 进行判断,直到一方的 next 为空
if cursor_a.val <= cursor_b.val:
cursor_res.next = ListNode(cursor_a.val)
cursor_a = cursor_a.next
else:
cursor_res.next = ListNode(cursor_b.val)
cursor_b = cursor_b.next
cursor_res = cursor_res.next
# 有一方的next的为空,就没有比较的必要了,直接把不空的一边加入到结果的 next 上
if cursor_a:
cursor_res.next = cursor_a
if cursor_b:
cursor_res.next = cursor_b
return dummy.next
def mergeKLists(self, lists: List[ListNode]) -> ListNode:
length = len(lists)
# 边界情况
if length == 0:
return None
if length == 1:
return lists[0]
# 分治
mid = length // 2
return self.merge(self.mergeKLists(lists[:mid]), self.mergeKLists(lists[mid:length]))
两两交换链表中的节点
链接:题目链接
题目描述
给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
示例:
给定 1->2->3->4, 你应该返回 2->1->4->3.
解题思路—用迭代思想,注意交换节点的顺序问题
这题可以用递归也可以用递归
迭代会比较好理解,我们要保存这一对中的两个节点,并且保存两个节点的pre节点
代码实现
class Solution:
def swapPairs(self, head: ListNode) -> ListNode:
dum = ListNode(-1)
dum.next = head
cur = head
pre = dum
while head and head.next:
A = head
B = head.next
pre.next = B
A.next = B.next
B.next = A
pre = A
head = A.next
return dum.next
## 迭代