Leetcode--Java--25. K 个一组翻转链表

题目描述

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

k 是一个正整数，它的值小于或等于链表的长度。

如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。
你可以设计一个只使用常数额外空间的算法来解决此问题吗？
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

样例描述

输入：head = [1,2,3,4,5], k = 2
输出：[2,1,4,3,5]

/**
 * 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; }
 * }
 */

思路

1. 本题思路与下面这题类似，
   Leetcode–Java–24. 两两交换链表中的节点
   
   
   三个步骤：
2. 先反转内部k个结点 （这个用的是双指针法，可参考下题）
   Leetcode–Java–206. Reverse Linked List
3. 修改第一个结点（一开始是虚拟结点）的指向
4. 修改反转后最后一个结点的指向，指向下一轮的待反转的结点

代码

class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
       if (head == null || head.next == null || k == 1) return head;
       ListNode dummy = new ListNode(-1);
       dummy.next = head;
       //通过工作指针为空来跳出循环
       ListNode p = dummy;
       while (true) {
           ListNode q = p;
           //先看是否后面k个都存在
           for (int i = 0; i < k && q != null; i ++ ) {
               q = q.next;
           }
           //没有满足k个都存在
           if (q == null) break;         
           ListNode a = p.next, b = a.next;   //两个翻转工作指针
            //将k个结点进行内部翻转，需要k - 1次翻转操作
            for (int i = 0; i < k - 1; i ++ ) {
                //记住b后面的，防止断链
                ListNode c = b.next;
                b.next = a;
                a = b;
                b = c;
            }
            //修改指向，接入到原链表，这里可以手画图对照更直观
          ListNode c = p.next;
          p.next = a;
          c.next = b;
          p = c;
       }
       return dummy.next;
    }
}