给定一个链表，旋转链表，将链表每个节点向右移动 k 个位置，其中 k 是非负数。
示例 1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例 2:
输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/rotate-list
解题思路
- 求出链表长度, 同时记录尾节点
- k %= length, 取模, k = length 也就是完整一圈, 相当于没移动
- k == 0直接返回
- 循环遍历找出新的头节点的上一个节点
- 将该节点的 next 记录为新头节点, 然后再将 next 置为 null 断开
- 将第一步记录的尾节点的 next 置为原头节点
代码
/**
* 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 || head.next == null || k == 0) {
return head;
}
ListNode cur = head;
ListNode tail; // 优化: 提前找出尾节点
// 求出链表长度
int length = 1;
while (cur.next != null) {
length++;
cur = cur.next;
}
tail = cur; // 记录尾节点
k %= length; // 原先用while去减去length, 傻逼
if (k == 0) {
return head; // k=0直接返回
}
ListNode newHead; // 新头节点
cur = head;
// 循环找出新的头节点的上一个节点
for (int i = 0; i < length - k - 1; i++) {
cur = cur.next;
}
newHead = cur.next; // 新头节点
cur.next = null; // 断开
tail.next = head; // 尾节点连上原来的头节点
return newHead;
}
}
