25. K 个一组翻转链表(Java实现)

力扣：25.K个一组翻转链表（困难）
参考：labuladong
题目：

答案：

/**
     * 反转区间 [a, b) 的元素，注意是左闭右开
     */
    ListNode reverse(ListNode a, ListNode b) {
        ListNode pre, cur, nxt;
        pre = null;
        cur = a;
        nxt = a;
        while (cur != b) {
            nxt = cur.next;
            cur.next = pre;
            pre = cur;
            cur = nxt;
        }
        return pre;// 返回反转后的头结点
    }

    ListNode reverseKGroup(ListNode head, int k) {
        if (head == null)
            return null;
        ListNode a, b; // 区间 [a, b) 包含 k 个待反转元素
        a = b = head;
        for (int i = 0; i < k; i++) {
            if (b == null) // 不足 k 个，不需要反转，base case
                return head;
            b = b.next;
        }
        ListNode newHead = reverse(a, b); // 反转前 k 个元素
        a.next = reverseKGroup(b, k); // 递归反转后续链表并连接起来
        return newHead;
    }

赶时间的朋友直接复制，力扣能够通过，下面开始是解析。

---

一、分析问题

递归：比如2个节点为一组反转链表，前2个节点反转，后面节点也是一条链表，而且规模比原来小，这个叫子问题。

直接递归调用 reverseKGroup(cur, 2)，因为子问题和原问题的结构完全相同，这就是所谓的递归性质。

而后将两个过程结果连接起来。

 

二、代码实现

反转整个链表：

// 反转以 a 为头结点的链表
ListNode reverse(ListNode a) {
    ListNode pre, cur, nxt;
    pre = null; cur = a; nxt = a;
    while (cur != null) {
        nxt = cur.next;
        // 逐个结点反转
        cur.next = pre;
        // 更新指针位置
        pre = cur;
        cur = nxt;
    }
    // 返回反转后的头结点
    return pre;
}

「反转以 a 为头结点的链表」其实就是「反转 a 到 null 之间的结点」，那么「反转 a 到 b 之间的结点」

只要更改函数签名，并把上面的代码中 null 改成 b 即可：

/** 反转区间 [a, b) 的元素，注意是左闭右开 */
ListNode reverse(ListNode a, ListNode b) {
    ListNode pre, cur, nxt;
    pre = null; cur = a; nxt = a;
    // while 终止的条件改一下就行了
    while (cur != b) {
        nxt = cur.next;
        cur.next = pre;
        pre = cur;
        cur = nxt;
    }
    // 返回反转后的头结点
    return pre;
}

现在我们迭代实现了反转部分链表的功能，接下来就按照之前的逻辑编写 reverseKGroup 函数即可：

ListNode reverseKGroup(ListNode head, int k) {
    if (head == null) return null;
    // 区间 [a, b) 包含 k 个待反转元素
    ListNode a, b;
    a = b = head;
    for (int i = 0; i < k; i++) {
        // 不足 k 个，不需要反转，base case
        if (b == null) return head;
        b = b.next;
    }
    // 反转前 k 个元素
    ListNode newHead = reverse(a, b);
    // 递归反转后续链表并连接起来
    a.next = reverseKGroup(b, k);
    return newHead;
}