#21_力扣打卡第21天(25)

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

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

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

进阶:

你可以设计一个只使用常数额外空间的算法来解决此问题吗?
你不能只是单纯的改变节点内部的值,而是需要实际进行节点交换。
 

示例 1:


输入:head = [1,2,3,4,5], k = 2
输出:[2,1,4,3,5]
示例 2:


输入:head = [1,2,3,4,5], k = 3
输出:[3,2,1,4,5]
示例 3:

输入:head = [1,2,3,4,5], k = 1
输出:[1,2,3,4,5]
示例 4:

输入:head = [1], k = 1
输出:[1]
提示:

列表中节点的数量在范围 sz 内
1 <= sz <= 5000
0 <= Node.val <= 1000
1 <= k <= sz

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group
 

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
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

思路:

看到题干,发现该题的解决通过两个部分,第一个部分是实现函数的翻转,有很多方法,第二个部分是对链表进行分组,然后对翻转的部分与其余部分进行拼接。

思考过后,发现第二个部分的实现比较简单,无非是设置一个计数器,每跳k次重置,还有关于拼接部分,也是将翻转后的链表的头和尾分别指向该指的连点。

翻转的方法:

(参考力扣评论区)

1.堆栈法:

通过将连点逐次放入栈中然后弹出,自然得到一个翻转好的链表。

2.尾插法:

举个例子:
1 2 3 4

2 3 4 1

3 4 2 1

4 3 2 1

即依次将头节点插入最开始的尾节点的后面,自然得到翻转链表。

3.交换法:

假如链表长度为偶数,我们只需将头和尾两两交换,每交换一次,头节点向后移动,尾节点向前移动,直到相等为止。

假如链表长度为奇数,则当两个指针所指为同一区域时退出。

4.模拟法:
代码:

class Solution:
    # 翻转一个子链表,并且返回新的头与尾
    def reverse(self, head: ListNode, tail: ListNode):
        prev = tail.next
        p = head
        while prev != tail:
            nex = p.next
            p.next = prev
            prev = p
            p = nex
        return tail, head

    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:
        hair = ListNode(0)
        hair.next = head
        pre = hair

        while head:
            tail = pre
            # 查看剩余部分长度是否大于等于 k
            for i in range(k):
                tail = tail.next
                if not tail:
                    return hair.next
            nex = tail.next
            head, tail = self.reverse(head, tail)
            # 把子链表重新接回原链表
            pre.next = head
            tail.next = nex
            pre = tail
            head = tail.next
        
        return hair.next

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/reverse-nodes-in-k-group/solution/k-ge-yi-zu-fan-zhuan-lian-biao-by-leetcode-solutio/
来源:力扣(LeetCode)

该方法摘自力扣,关于翻转部分,用了四个指针,为了避免考虑复杂的情况,用了四个指针,外加一个dummy点。


版权声明:本文为m0_56215962原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。