[12][01][61] 旋转链表

1. 题目描述

给你一个链表的头节点 head, 旋转链表, 将链表每个节点向右移动 k 个位置

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

输入：head = [0, 1, 2], k = 4
输出：[2, 0, 1]

2. 解题思路

• 当链表 head 为空或只有一个结点的情况, 直接返回 head
• 当向右移动 k 个位置的 k = 0, 不需要处理直接返回 head
• 计算出链表节点总数 length 后, 需要 k = k % length, 用于处理 k > length 场景, 还需要判断 k == 0, 直接返回 head

此题的关键点有三个

• 定义一个最终结果链表 ListNode result = new ListNode(-1, head), 将 result.next 指向倒数第 k 个结点
• 将倒数第 k + 1 个结点与倒数第 k 个结点断开
• 将最后一个结点的 next 指向第一个结点 head

public ListNode rotateRight(ListNode head, int k) {
    // 如果 head 为空或只有一个结点,无需移动结点直接返回 head
    // 如果 k == 0, 无需移动结点, 直接返回 head
    if (null == head || head.next == null || k == 0) {
        return head;
    }

    // 定义最终结果链表, 加上头结点
    ListNode result = new ListNode(-1, head);
    ListNode last = head;

    // 遍历链表得到链表元素个数
    int length = 1;
    while (last.next != null) {
        last = last.next;
        length++;
    }

    // 处理 k > length 场景
    k = k % length;
    // 处理 k 是 length 的整数倍, 无需处理head,直接返回
    if (k == 0) {
        return result.next;
    }

    ListNode temp = head;
    // 找到倒数第 k + 1 个结点
    for (int i = 0; i < length - k - 1; i++) {
        temp = temp.next;
    }
    // 将结果链表 result.next 指向倒数第 k 个结点
    result.next = temp.next;
    // 将倒数第 k + 1 个结点与倒数第 k 个结点断开
    temp.next = null;
    // 将最后一个结点的 next 指向第一个结点 head
    last.next = head;
    // 返回结果
    return result.next;
}