K个一组翻转链表
题目描述:
给出一个链表,每 k 个节点一组进行翻转,并返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍,那么将最后剩余节点保持原有顺序。
示例
给定这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明 :
- 你的算法只能使用常数的额外空间。
- 你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
解题思路:
主要难点在对K值的判断和链表头插法
链表头插法:
在头节点的后面进行插入操作,后一个插入进来的值,在前一个插入进来的值与头节点之间。
首先创建临时头结点
pre
,用指针tail
指向它-
开始循环遍历,需要先对链表向后查找k个节点
while p is not None and n > 0: p = p.next n -= 1
如果在查找
k
个节点的过程中遇到None
,则说明后面的节点不够k个节点则直接跳出即可如果还在范围内,则进行链表的头插法,假设对于链表
Head->A->B->C
进行排序,假设此时已经有Head->A
创建
q
指针指向B,创建t
指针指向C(),接下来将B指向A()然后将
Head
指向B(),最后将q
指针指向C(),开始下一次的头插
Python源码:
# Definition for singly-linked list.
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
pre = ListNode(-1)
tail = pre
q = head
while q is not None:
# 向后查找k个节点
n = k
p = q
while p is not None and n > 0:
p = p.next
n -= 1
# 如果在查找k个节点的过程中遇到None,则说明
# 后面的节点不够k个节点则直接跳出即可
if n > 0:
tail.next = q
break
# 将这K个节点以头插法插入
end = q
while q != p:
t = q.next
q.next = tail.next
tail.next = q
q = t
tail = end
return pre.next
欢迎关注我的github:https://github.com/UESTCYangHR