链表算法三K个一组链表翻转(力扣25题)

力扣25题

题目

给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。 k 是一个正整数，它的值小于或等于链表的长度。 如果节点总数不是 k
的整数倍，那么请将最后剩余的节点保持原有顺序。 来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reverse-nodes-in-k-group

解题思路

假设我们现在有一个链表

长成图片上这个样子,每3(K = 3)个一组,第7个节点自己一组,无需翻转。

第一步: 翻转第一组

让1指向NULL,2指向1,3指向2,最后再让1指向4,翻转后应该就是下图样子

第二步:翻转第二组

和第一步一样翻转,让6指向5,5指向4,4指向7,如下图所示

第三步:改正上一次错误的指向

1节点指向4节点的时候4节点所在组还没有进行反转,现在进行完了反转那1节点就应该指向6节点
用代码的语言来说就是===================
第一次翻转 :
开始节点 ==> start = 1节点
结束节点 ==> end = 3节点
开始节点的下一个节点 ==> start1.next = 4节点(不指向4节点的话就是start1.next就是NULL)
第二次翻转:
开始节点 ==> start = 4节点
结束节点 ==> end = 6节点
反转后连接下一组 ==> start1.next = 7节点
重点来了!!
第二次翻转后第一次翻转的<开始节点的下一个节点>应该指向第二次翻转的<结束节点>,原因是因为第二组反转了,掉了ge个所以应该指向end
如图:

第四步:翻转第N组

第二步第三步重复操作即可,写个循环

代码如下

class ListNode {
    int val;
    ListNode next;

    ListNode() {
    }

    ListNode(int val) {
        this.val = val;
    }

    ListNode(int val, ListNode next) {
        this.val = val;
        this.next = next;
    }
}

public class Kreversal {

    public ListNode reverseKGroup(ListNode head, int k) {
        // 上来直接获取end节点位置
        ListNode end = getKGroupEnd(head, k);
        // 如果end节点为空代表链表长度小于k直接返回即可
        if (end == null) {
            return head;
        }
        // 定义start变量
        ListNode start = head;
        // 第一次反转
        reversal(start, end);
        // 让head指向end位置
        head = end;

        // 定义lastEnd记录上一次结束的位置(直接.next = 这次start的位置)
        ListNode lastEnd = start;

        while (lastEnd.next != null) {
            // 获取本次处理链表的起始位置
            start = lastEnd.next;
            // 获取本次处理链表的结束位置
            end = getKGroupEnd(start, k);
            if(end == null){
                return head;
            }
            // 反转链表
            reversal(start, end);
            // 当前这一段链表反转后,上一次链表结束位置还指向本次链表的start位置,但是现在start和end反转了,所以应该指向end
            lastEnd.next = end;
            // lastEnd指向本次循环结束位置
            lastEnd = start;
        }
        return head;
    }


    /**
     * 获取K获取end指针
     *
     * @param start start起始指针
     * @param k     K
     * @return 结束指针
     */
    public ListNode getKGroupEnd(ListNode start, int k) {
        while (--k > 0 && start != null) {
            start = start.next;
        }
        return start;
    }

    public void reversal(ListNode start, ListNode end) {
        // 让end指向自己下一个节点
        end = end.next;
        // 定义pre
        ListNode pre = null;
        // 定义当前节点变量
        ListNode cur = start;
        // 定义下一个节点元素
        ListNode next;
        while (cur != end) {
            next = cur.next;
            cur.next = pre;
            pre = cur;
            cur = next;
        }
        start.next = end;
    }
}

最后放一个力扣提交截屏