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

题目:
K 个一组翻转链表:给你一个链表,每 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:
        if not head:
            return None
        if k <= 1:
            return head
        myStack = []
        myNewList = ListNode(0)
        myNewList.next = head
        myNewHead = myNewList
        num = k 
        while myNewHead and myNewHead.next:
            node = myNewHead.next
            while num > 0 and node:
                myStack.append(node)
                node = node.next
                num -= 1
            section_next = myStack[-1].next
            if num == 0:
                for index in range(len(myStack)):
                    myNewHead.next = myStack.pop()
                    myNewHead = myNewHead.next
            else:
                break
            myNewHead.next = section_next
            num = k 
        return myNewList.next

猜你喜欢

转载自www.cnblogs.com/zhuozige/p/12824550.html