leetcode 25 k 个一组翻转列表

题目描述

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。

k 是一个正整数，它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

 

示例：

给你这个链表：1->2->3->4->5

当 k = 2 时，应当返回: 2->1->4->3->5

当 k = 3 时，应当返回: 3->2->1->4->5

 

说明：

你的算法只能使用常数的额外空间。 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

 

解题思路

对于翻转数组类问题，除了遍历指针方法外，堆栈结构是一个不错的选择。

1、k个一组把节点放到堆栈中去

2、逐个pop堆栈中的节点

3、处理堆栈中剩余的节点

class Solution: def reverseKGroup(self, head: ListNode, k: int) -> ListNode: stack = [] final = ListNode(0) hair = final # 遍历整个链表 while head: # k个一组将链表中的节点入栈 # 当不满k个节点时，循环结束，堆栈中是剩余的节点 if len(stack) < k: stack.append(head) head = head.next # k个节点进行翻转操作 else: while stack: hair.next = stack[-1] hair = hair.next stack.pop(-1) # 如果堆栈中有链表中剩余不足k个节点 # 分两种情况进行处理 # 不足k个，不进行翻转操作 # 大于等于k个，进行翻转操作 输入 [1,2] 2, while下的else是不执行的， # 此时stack的长度是等于k的，所以也需要考虑大于等于k的情况。 if len(stack) < k: while stack: hair.next = stack[0] hair = hair.next stack.pop(0) else: while stack: hair.next = stack[-1] hair = hair.next stack.pop(-1) hair.next = None return final.next

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group