【Leetcode】61. 旋转链表

题目描述

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

题解

旋转链表 = 找倒数第k个节点 + 翻转前后链表

执行用时：0 ms, 在所有 Java 提交中击败了100.00%的用户

内存消耗：37.8 MB, 在所有 Java 提交中击败了62.16%的用户

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    int len = 0;
    int rotate = 0;

    public ListNode rotateRight(ListNode head, int k) {
        if (head == null || head.next == null || k == 0)
            return head;
        ListNode count = head;
        while (count != null) {  // 计算链表长度len
            count = count.next;
            len++;
        }
        if (len <= k) k = k % len;  // 如果k<=len，先消掉进位
        if (k == 0) return head;  // 如果k == 0，说明不需要旋转，直接返回

        // 旋转链表 = 找倒数第k个节点 + 翻转前后链表
        ListNode pre = head;
        ListNode cur = head; 
        for (int i = 0; i < k; i++) {  // 先创建一个长度为k的窗口，pre是窗口左边界,cur是窗口又边界
            cur = cur.next;
        }
        while (cur.next != null) {  // 窗口遍历到链表尾，则窗口左边界位置就是倒数第k个节点位置
            pre = pre.next;
            cur = cur.next;
        }

        // 翻转前后链表
        ListNode newHead = pre.next;
        pre.next = null;
        cur.next = head;
        return newHead;
    }
}