LeetCode 25. K 个一组翻转链表
25. K 个一组翻转链表
Difficulty: 困难
给你一个链表,每 _k _个节点一组进行翻转,请你返回翻转后的链表。
_k _是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 _k _的整数倍,那么请将最后剩余的节点保持原有顺序。
示例:
给你这个链表:1->2->3->4->5
当 _k _= 2 时,应当返回: 2->1->4->3->5
当 _k _= 3 时,应当返回: 3->2->1->4->5
说明:
- 你的算法只能使用常数的额外空间。
- 你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
Solution
不同于206题反转链表的整个部分,本题的要求是反转链表的一部分,要求每 k 个节点一组进行翻转。
首先我们需要获取链表的长度length
,如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序,比如长度为5的链表,3个一组反转,此时只需要反转前三个节点即可,剩余的2个保持原样。
遇到需要额外空间的链表题,需要初始化dummy
和p
变量,p
用于完成实际的节点交换。k个一组的节点交换过程为不断地把当前节点cur
前面的节点放到cur
的身后,实际的操作过程如下图所示:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
if not head:
return head
# 获取链表的长度length
helpNode = head
length = 0
while helpNode:
helpNode = helpNode.next
length += 1
dummy = p = ListNode(-1)
p.next = head
t = length // k # 计算需要反转的链表组数
for i in range(t):
cur = p.next
for _ in range(k-1):
c_next = cur.next
cur.next = c_next.next # ①
c_next.next = p.next # ②
p.next = c_next # ③
p = cur
return dummy.next
相关题目: