题目描述
给你一个链表的头节点 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/rotate-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
虽然说是旋转链表,但是实际上可以把它当作一个回环链表,然后在特定地方把链断开就是答案
- 如果链表为空直接返回空链表即可
- 遍历链表,确定链表长度 len,然后把链表最后节点的next指向链表开头,形成回环
- 移动次数 k 对 len 取模,重新确定 k
- 再遍历一遍链表,在相应节点(与len-k相关,看代码的循环条件)处记录下一节点为新head,然后将head前一节点的next指向null,那么最终需要的链表就是以新head为头节点的链表
代码详解
/**
* 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;
}
ListNode temp = head; // 保存一下头节点
int len = 0; // 记录链表长度
while(head != null) {
++len;
if(head.next == null) {
head.next = temp; // 链表最后节点的next指向头节点,形成大回环
break;
}
head = head.next;
}
head = temp; // 再从头节点进行最后一次遍历
k = k % len; // 取模,重新确定k的最小等效值
for(int i = 1; i < len-k; ++i) {
temp = temp.next; // 遍历到第len-k个节点
}
head = temp.next; // 新的头节点为temp.next
temp.next = null; // 再把temp后的链断开即可
return head;
}
}
注意点
- 不要被“旋转”、“移动”蒙蔽双眼,此题就是一个链表环然后在相应位置处断开的一道题