美文网首页
[LeetCode]25、K个一组翻转链表

[LeetCode]25、K个一组翻转链表

作者: 河海中最菜 | 来源:发表于2019-07-28 08:45 被阅读0次

    题目描述

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

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

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

    示例 :

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

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

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

    说明 :

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

    思路:

    既然是翻转,利用栈来翻转。每次放入栈中K个,否则停止翻转

    # 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:
            dummy = ListNode(0)
            p = dummy
            while True:
                count = k
                stack = []
                tmp = head
                while count and tmp:
                    stack.append(tmp)
                    tmp = tmp.next
                    count -= 1
                # 注意,目前tmp所在k+1位置
                # 说明剩下的链表不够k个,跳出循环
                if count:
                    p.next = head
                    break
                # 翻转操作
                while stack:
                    p.next = stack.pop()
                    p = p.next
                # 与剩下链表连接起来
                p.next = tmp
                head = tmp
    
            return dummy.next
    
    AC25

    相关文章

      网友评论

          本文标题:[LeetCode]25、K个一组翻转链表

          本文链接:https://www.haomeiwen.com/subject/zesyrctx.html