LeetCode25 K个一组翻转链表

题目

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

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

如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

进阶：
你可以设计一个只使用常数额外空间的算法来解决此问题吗？
你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。

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

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

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

 示例 4： 
输入：head = [1], k = 1
输出：[1]

 提示： 
 列表中节点的数量在范围 sz 内 
 1 <= sz <= 5000 
 0 <= Node.val <= 1000 
 1 <= k <= sz 

方法

模拟法

• 时间复杂度：O(n)，n为链表节点个数
• 空间复杂度：O(1)

class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode newHead = new ListNode();
        newHead.next = head;
        ListNode pre = newHead;
        while (head!=null){
            ListNode tail = pre;
            for(int i=0;i<k;i++){
                tail = tail.next;
                if(tail==null){ //判断一定要写在后面，因为后面会掉tail.next,写前面如果插好tail为null会拦不住
                    return newHead.next;
                }
            }
            ListNode next = tail.next;
            ListNode[] res = reverse(pre.next,tail);
            head = res[0];
            tail = res[1];
            pre.next = head;
            tail.next = next;
            pre = tail;
        }
        return newHead.next;
    }
    private ListNode[] reverse(ListNode head,ListNode tail){
        //包含tail
        ListNode pre = tail.next,node = head;
        while (pre != tail){
            ListNode next = node.next;
            node.next = pre;
            pre = node;
            node = next;
        }
        return new ListNode[]{tail,head};
    }
}

模拟法+变量简化

reverse方法翻转时已经把尾巴连上了，只需要每次把头接上就可以

• 时间复杂度：O(n)，n为链表节点个数
• 空间复杂度：O(1)

class Solution {
    public ListNode reverseKGroup(ListNode head, int k) {
        ListNode newHead = new ListNode();
        newHead.next = head;
        ListNode pre = newHead;
        while (pre!=null){
            ListNode tail = pre;
            for(int i=0;i<k;i++){
                tail = tail.next;
                if(tail==null){
                    return newHead.next;
                }
            }
            ListNode[] res = reverse(pre.next,tail);
            pre.next =  res[0];
            pre = res[1];
        }
        return newHead.next;
    }
    private ListNode[] reverse(ListNode head,ListNode tail){
        //包含tail，pre初始值为next是把尾巴连上
        ListNode pre = tail.next,node = head;
        while (pre != tail){
            ListNode next = node.next;
            node.next = pre;
            pre = node;
            node = next;
        }
        return new ListNode[]{tail,head};
    }
}