链表中的节点每k个一组翻转——NC.50

题目描述

$将给出的链表中的节点每k个一组翻转，$ $返回翻转后的链表如果链表中的节点数不是k的倍数，$ $将最后剩下的节点保持原样$
$你不能更改节点中的值，只能更改节点本身。$ $要求空间复杂度O(1)$
$例如：$ $给定的链表是1\to 2\to 3\to 4\to 5$
$对于k=2,你应该返回2\to 1\to 4\to 3\to 5$
$对于k=3,你应该返回3\to 2\to 1\to 4\to 5$

大体思路：
先求出链表的长度，这样可以计算处后续要反转多少次。
再根据k值反转链表。

反转链表过程的代码很巧妙，这里用图详细解释一下：

import java.util.*;

/*
 * public class ListNode {
 *   int val;
 *   ListNode next = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param head ListNode类 
     * @param k int整型 
     * @return ListNode类
     */
    public ListNode reverseKGroup (ListNode head, int k) {
        // write code here
        if (head == null || head.next == null || k < 2) return head;
        ListNode res = new ListNode(-1);
        res.next = head;
        ListNode pre = res, cur = head, tmp;
        int len = 0;
        while (head != null) {
            len++;
            head = head.next;
        }
        
        for (int i = 0; i < len / k; i++) {
            for (int j = 1; j < k; j++) {    // 因为cur = head, 已经算一个
                tmp = cur.next;
                cur.next = tmp.next;
                tmp.next = pre.next;
                pre.next = tmp;
            }
            pre = cur;
            cur = pre.next;
        }
        return res.next;
    }
}