leetcode 25. K 个一组翻转链表(python3)60ms

题目描述:

给你一个链表,每 k 个节点一组进行翻转,请你返回翻转后的链表。
k 是一个正整数,它的值小于或等于链表的长度。
如果节点总数不是 k 的整数倍,那么请将最后剩余的节点保持原有顺序。

示例:

给你这个链表:1->2->3->4->5
当 k = 2 时,应当返回: 2->1->4->3->5
当 k = 3 时,应当返回: 3->2->1->4->5
说明:
你的算法只能使用常数的额外空间。
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。

解题思路:

# 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:
        output = ListNode(0)
        ptr_node = output
        read_node = head
        while read_node:
            cnt = k
            tmp = []
            while read_node and cnt:
                tmp.append(read_node)
                read_node = read_node.next
                cnt = cnt - 1
            if cnt:
                ptr_node.next = head
                return output.next
            while tmp:
                ptr_node.next = tmp.pop()
                ptr_node = ptr_node.next
                ptr_node.next = None
            head = read_node
        return output.next

算法性能:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/lanmengyiyu/article/details/108538908
今日推荐