面试-K个一组翻转链表

题目

给你链表的头节点 head ，每 k 个节点一组进行翻转，请你返回修改后的链表。

k 是一个正整数，它的值小于或等于链表的长度。如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

思路

class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        //pre和end分别表示待翻转链表的起始位置的上一个节点和结束位置
        ListNode pre = dummy;
        ListNode end = dummy;
        
        while (end.next != null) {//因为需要定位到end的后继节点，所以判断用end.next
            for (int i = 0; i < k && end != null; i++) 
                end = end.next;//找到需要翻转的k个链表的结束位置，dummy->1->2->3->4->5 若k为2，循环2次，end指向2
                
            if (end == null) break;//如果end == null，即需要翻转的链表的节点数小于k，不执行翻转。
            
            ListNode start = pre.next;//记录待翻转链表的头节点，pre.next
            ListNode tmp = end.next;//记录待翻转链表的下一个节点，方便连接后面链表
            end.next = null;//断开链表
            pre.next = reverse(start);//翻转链表,pre.next指向翻转后的链表。1->2 变成2->1。 dummy->2->1
            start.next = tmp;//翻转后头节点变到最后。通过start.next把断开的链表重新链接。
            pre = start;//将pre换成下次要翻转的链表的头结点的上一个节点。即start
            end = pre;//end也重新开始遍历，指向待翻转节点的结束位置
        }
        return dummy.next;
    }

    private ListNode reverse(ListNode head) {
        ListNode pre = null;
        ListNode curr = head;
        while (curr != null) {
            ListNode next = curr.next;
            curr.next = pre;
            pre = curr;
            curr = next;
        }
        return pre;
    }
}