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个保持原样。

遇到需要额外空间的链表题,需要初始化dummyp变量,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

相关题目:

LeetCode 143. 重排链表
LeetCode 206. 反转链表
LeetCode 92. 反转链表 II

posted @ 2020-12-26 20:26  swordspoet  阅读(46)  评论(0编辑  收藏  举报