首页 > 其他 > 详细

LeetCode 25. K 个一组翻转链表

时间:2020-07-11 19:59:18      阅读:56      评论:0      收藏:0      [点我收藏+]

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

示例:
给你这个链表:1->2->3->4->5
当?k?= 2 时,应当返回: 2->1->4->3->5
当?k?= 3 时,应当返回: 3->2->1->4->5

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

class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        if head is None:
            return None
        cur = head
        length = 0
        while cur:
            cur = cur.next
            length += 1
        
        n,_ = divmod(length,k)
        for i in range(n):
            head = self.reverseBetween(head,i*k+1,i*k+k)
        return head 

    def reverseBetween(self, head: ListNode, m: int, n: int) -> ListNode:
        if head is None:
            return head
        pre_head = ListNode(None)
        pre_head.next = head
        pre = pre_head
        for _ in range(m-1):
            pre = pre.next

        cur = pre.next
        for _ in range(n-m):
            tmp = cur.next
            cur.next = tmp.next
            tmp.next = pre.next
            pre.next = tmp
        return pre_head.next

LeetCode 25. K 个一组翻转链表

原文:https://www.cnblogs.com/sandy-t/p/13285074.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!