链表中的节点每k个一组翻转

链表中的节点每k个一组翻转

题目描述

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

代码

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 == 1) return head;//笨死了我，居然写成 head.next != null，找了下面程序半天错误，哭
        
        
        //上一链条最后一个节点
        ListNode pre_tail = null;
        //当前链的头节点
        ListNode cur_head = head;
        //当点链的尾结点
        ListNode cur_tail = head;
        
        int count = k-1;
        while(cur_tail!=null){
            //获取当前链的尾结点
            if( count!=0 ){
                cur_tail = cur_tail.next;
                count--;
            }
            if(count == 0 && cur_tail!=null){
                //反转当前链表
                reverseK(cur_head, cur_tail);
                //连接当前链和上一条链
                if(pre_tail==null){
                    head = cur_tail;
                }else{
                    pre_tail.next = cur_tail;
                }
                //下一轮转换
                pre_tail = cur_head;
                cur_head = pre_tail.next;
                cur_tail = cur_head;
                count = k-1;
            }
            //count不为零且cur_tail=null，直接退出，k余数后的节点不做操作
        }
        return head;
    }
    
    public void reverseK(ListNode cur_head, ListNode cur_tail){
        //为了获取下一链条的开始节点
        ListNode head_next = cur_head;
        //前后节点
        ListNode p1 = cur_head;
        ListNode p2 = cur_head.next;
        //
        ListNode next_head = cur_tail.next;
        
        while( p2!=next_head && p2!=null ){
            head_next.next = p2.next;
            p2.next = p1;
            p1 = p2;
            //下一轮
            p2 = head_next.next;
        }
    }
}