【经典算法】LeetCode25:K 个一组翻转链表(Java/C/Python3,Hard)

原创

已于 2024-04-07 09:10:18 修改 · 1.3k 阅读

标签

#K个一组翻转链表 #链表翻转 #力扣

于 2024-04-07 09:03:34 首次发布

本文详细介绍了如何使用递归和迭代原地反转的方法解决LeetCode25题（K个节点一组翻转链表），包括Java、C语言和Python实现，以及它们的时间复杂度和空间复杂度分析。

#算法

题目描述

给你链表的头节点 head ，每 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]
提示：
链表中的节点数目为 n
1 <= k <= n <= 5000
0 <= Node.val <= 1000
进阶：你可以设计一个只用 O(1) 额外内存空间的算法解决此问题吗？

原题：LeetCode 25

思路及实现

方式一：递归

思路

其大致过程可以分解为

找到待翻转的k个节点（注意：若剩余数量小于 k 的话，则不需要反转，因此直接返回待翻转部分的头结点即可）。
对其进行翻转。并返回翻转后的头结点（注意：翻转为左闭又开区间，所以本轮操作的尾结点其实就是下一轮操作的头结点）。
对下一轮 k 个节点也进行翻转操作。
将上一轮翻转后的尾结点指向下一轮翻转后的头节点，即将每一轮翻转的k的节点连接起来。

示意图

代码实现

Java 版本

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
   
   
    /**
     * 反转链表中每个大小为 k 的连续节点的子链表
     * @param head 当前子链表的头节点
     * @param k 指定的连续节点个数
     * @return 反转后的链表的头节点
     */
    public ListNode reverseKGroup(ListNode head, int k) {
   
   
        if (head == null || head.next == null) {
   
   
            return head;
        }
        ListNode tail = head;
        for (int i = 0; i < k; i++) {
   
   
            // 如果剩余数量小于 k，则不需要反转。
            if (tail == null) {
   
   
                return head;
            }
            tail = tail.next;
        }
        // 反转前 k 个元素
        ListNode newHead = reverse(head, tail);
        // 下一轮的开始的地方就是 tail
        head.next = reverseKGroup(tail, k);
        return newHead;
    }

    /**
     * 反转链表中左闭右开区间的节点
     * @param head 左闭区间的头节点
     * @param tail 右开区间的尾节点
     * @return 反转后的链表的头节点
     */
    private ListNode reverse(ListNode head, ListNode tail) {
   
   
        ListNode pre = null;
        ListNode next = null;
        while (head != tail) {
   
   
            next = head.next;
            head.next = pre;
            pre = head;
            head = next;
        }
        return pre;
    }
}

说明：
reverseKGroup() 方法用于将链表每个大小为 k 的子列表进行反转。
如果输入的头节点 head 或其下一个节点为空，则无需翻转，直接返回头节点。
使用 tail 指针找到当前子列表的结束节点（即当前子列表的下一组的开始节点）。
如果剩余节点数量不足 k，则无需进行翻转，直接返回头节点。
调用 reverse() 方法反转当前子列表，并得到翻转后的新的头节点 newHead。
通过递归调用 reverseKGroup() 方法，将下一轮的开始位置 tail 作为参数传入。
将当前子列表的头节点 head 的 next 指针指向下一轮的结果，连接翻转后的下一组子列表。
返回翻转后的新的头节点 newHead。

C 语言版本

struct ListNode {
   
   
    int val;
    struct ListNode *next;
};

/**
 * 反转以头节点head开始，尾节点为tail前一个节点的链表
 * 返回反转后的链表的头节点
 */
struct ListNode* reverse(struct ListNode* head, struct ListNode* tail) {
   
   
    struct ListNode* prev = NULL;
    struct ListNode* curr = head;
    struct