题解 | #链表中的节点每k个一组翻转#直来直去的思路,推荐

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

https://www.nowcoder.com/practice/b49c3dc907814e9bbfa8437c251b028e

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) {
        // 先处理特殊情况
        if (head == null || head.next == null || k == 1) return head;
        //得到链表的长度
        int length = getLength(head);
        if (k > length) {
            return head;
        }
        //处理常规情况
        int groupCount = length / k;
        //最后一组是否满k个
        boolean isLastGroupLessThanK = false;
        if (length % k != 0) {
            groupCount += 1;
            isLastGroupLessThanK = true;
        }
        ListNode newHead = null;
        //用来记录上一组的尾部
        ListNode lastTail  = null;
        for (int i = 0; i < groupCount; i++) {
            //第一组和最后一组要特殊处理
            if (i == 0) {
                //第一组会得到头部
                //翻转k次
                ListNode pre = null;
                lastTail = head;
                for (int j = 0; j < k; j++) {
                    ListNode next = head.next;
                    head.next = pre;
                    pre = head;
                    head = next;
                }
                //执行到这里时 pre 是新头部,head 是下一组的头部
                newHead = pre;
            } else if (i == groupCount - 1 && isLastGroupLessThanK) {
                //最后一组 并且是不够K个要特殊处理
                lastTail.next = head;
            } else {
                ListNode pre = null;
                ListNode willTail = head;
                for (int j = 0; j < k; j++) {
                    ListNode next = head.next;
                    head.next = pre;
                    pre = head;
                    head = next;
                }
                //执行到这里后 pre 是当前组的头部 需要和上一组进行拼接
                lastTail.next = pre;
                lastTail = willTail;
            }
        }
        return newHead;
    }

    private int getLength(ListNode head) {
        int length = 0;
        ListNode temp = head;
        while (temp != null) {
            length ++;
            temp = temp.next;
        }
        return length;
    }
}

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-08 12:10
点赞 评论 收藏
分享
05-29 20:34
门头沟学院 C++
KarlAllen:得做好直接春招的准备。学历差的话,一是面试要求会比学历好的严格不少,二是就算面试通过了也会被排序。总之暑期和秋招对于学历差的就是及其不友好
无实习如何秋招上岸
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务