旋转链表

问题描述

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


示例 1：

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

示例 2：

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

提示：

    链表中节点的数目在范围 [0, 500] 内
    -100 <= Node.val <= 100
    0 <= k <= 2 * 109

Java代码

/**
 * 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 {
    public ListNode rotateRight(ListNode head, int k) {
        if (head == null) {
            return head;
        }
        // 当k = 0  或者只有一个元素的时候，都直接返回head即可
        if (k == 0 || head.next == null) {
            return head;
        }

        // 计算出链表的长度
        int len = 0;
        ListNode p = head;
        ListNode tail = null;
        while (p != null) {
            len++;
            if (p.next == null) {
                tail = p;
                break;
            }
            p = p.next;
        }

        if (k % len == 0) {
            return head;
        }

        // 变为环
        tail.next = head;

        // 计算出移动后的尾节点(下标从0开始)
        int x = len - k % len;

        while (x > 0) {
            tail = tail.next;
            x--;
        }
        // 找到尾节点，尾节点前面一个就是头节点
        ListNode newHead = tail.next;
        // 将尾节点的next设置为null
        tail.next = null;
        return newHead;
    }
}

leetcode链接