牛客网高频算法题系列-BM3-链表中的节点每k个一组翻转

牛客网高频算法题系列-BM3-链表中的节点每k个一组翻转

题目描述

将给出的链表中的节点每 k 个一组翻转,返回翻转后的链表如果链表中的节点数不是 k 的倍数,将最后剩下的节点保持原样你不能更改节点中的值,只能更改节点本身。

原题目见:BM3 链表中的节点每k个一组翻转

解法一:递归

使用递归的方式处理每k个一组的结点,具体处理方式如下:

  • 首先用tail记录每次翻转的尾部,就是第k+1个结点,从头结点开始,遍历链表,找到第k+1个结点,如果当前结点数不够k个,则不用反转直接返回;
  • 然后从链表头开始遍历,翻转结点并记录反转后的表头pre和表尾结点head;
  • 然后递归处理从tail开始的后面的链表结点,并将head指向递归处理后的链表;
  • 返回新的表头即为反转后的新的链表的表头结点。

代码

public class Bm003 {
    /**
     * 递归
     *
     * @param head ListNode类
     * @param k    int整型
     * @return ListNode类
     */
    public static ListNode reverseKGroup(ListNode head, int k) {
        // 找到每次翻转的尾部
        ListNode tail = head;
        // 遍历k次到尾部
        for (int i = 0; i < k; i++) {
            // 如果不足k到了链表尾部,直接返回,不翻转
            if (tail == null) {
                return head;
            }
            tail = tail.next;
        }
        // 翻转时需要的前序和当前节点
        ListNode pre = null, cur = head;
        // 在到达当前段尾部节点前
        while (cur != tail) {
            ListNode temp = cur.next;
            cur.next = pre;
            pre = cur;
            cur = temp;
        }
        // 当前尾部指向下一段要翻转的链表
        head.next = reverseKGroup(tail, k);
        return pre;
    }

    public static void main(String[] args) {
        ListNode head = ListNode.testCase2();
        System.out.println("翻转之前");
        ListNode.print(head);

        // 当k=2时
        ListNode newHead = reverseKGroup(head, 2);
        System.out.println("当k=2时,翻转之后");
        ListNode.print(newHead);

        ListNode head1 = ListNode.testCase2();
        // 当k=3时
        ListNode newHead2 = reverseKGroup(head1, 3);
        System.out.println("当k=3时,翻转之后");
        ListNode.print(newHead2);
    }
}



相信坚持的力量!

你可能感兴趣的:(牛客网高频算法题系列-BM3-链表中的节点每k个一组翻转)