LeetCode k个一组翻转链表

k个一组翻转链表

题目描述

给出一个链表，每 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:  
        if not head: return  
        
        next_list = [head]
        for i in range(k-1):
            if next_list[i].next: next_list.append(next_list[i].next)
            else : break
        
        if len(next_list) < k: return head
            
        res = next_list[-1]
        next_list[0].next = self.reverseKGroup(next_list[-1].next, k)
        for i in range(k-1):
            index = k-1-i
            next_list[index].next = next_list[index-1] 

        return res

方法二：

   class Solution:
    def reverseKGroup(self, head: ListNode, k: int) -> ListNode:  
        if not head: return  
        
        temp = head
        tail = head
        count = 1
        
        # 检查是否够长
        while tail.next and count<k:
            tail = tail.next
            count+=1
        
        if count == k:
            tail.next = self.reverseKGroup(tail.next, k)
            while temp!=tail:
                p = tail.next
                tail.next = temp
                temp = temp.next
                tail.next.next = p
            return tail
        else:
            return head