Skip to the content.

leetcode [25] K 个一组翻转链表


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

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

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

示例:

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

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

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

说明:

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

tail指针从头遍历,计数k,得到需要翻转链表的头和尾。为了方便新建一个头节点和尾节点,其next指向head。每次需要翻转的链表头部为新建尾节点的下一个。

class Solution:
    def reverse(self, head, end):
        prev, cur, post = None, head, None
        while cur != end:
            post = cur.next
            cur.next = prev
            prev = cur
            cur = post
        return prev, head

    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        if not head or k <= 0: return head
        tail = head
        final_head = ListNode(0)
        final_head.next = head
        final_tail = final_head
        count = 0
        while tail:
            count += 1
            tail = tail.next
            if count == k:
                h, t = self.reverse(final_tail.next, tail)
                final_tail.next = h
                t.next = tail
                final_tail = t
                count = 0
        return final_head.next