25. K 个一组翻转链表

cccc楚染rrrr2023-12-27 11:26

题解参考：https://leetcode.cn/problems/reverse-nodes-in-k-group/solutions/10416/tu-jie-kge-yi-zu-fan-zhuan-lian-biao-by-user7208t/

设置dummy虚拟头节点，pre为待翻转部分的前驱（用于连接），end为待翻转部分中的结尾节点（用k定），next为待翻转部分的后续节点（用于连接），bg为翻转前的翻转部分的开始节点，进入reverse后，会有cur节点作为副本，因此bg的位置不会变，翻转后，bg所在节点变成了结尾，将其和next相连，reverse返回的节点（新开始节点）和pre相连。
pre指向bg，作为下一个待翻转部分的前驱。
end指向pre，从pre出发数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 {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummy = new ListNode(0, head);
        ListNode pre = dummy, end = dummy;
        while (end.next != null) {
            for (int i = 0; i < k && end != null; ++i) end = end.next; 
            // 注意只要访问next节点，就要检查这个节点是否为空
            if (end == null) break;
            ListNode bg = pre.next;
            ListNode nxt = end.next;
            end.next = null;
            pre.next = reverse(bg);
            bg.next = nxt;
            pre = bg;
            end = pre;
        }
        return dummy.next;
    }
    private ListNode reverse(ListNode head) { 
        ListNode pre = null;
        ListNode cur = head;
        while (cur != null) {
            ListNode nxt = cur.next;
            cur.next = pre;
            pre = cur;
            cur = nxt;
        }
        return pre;
    }
}