给你一个链表的头节点 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
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(!head){
return nullptr;
}
ListNode* rotateHead=head;
ListNode* rotateTail=head;
int size=1;
while(rotateTail->next){
size++;
rotateTail=rotateTail->next;
}
if(size==1||k%size==0||k==size) return head;
k=k%size;
k=size-k-1;
while(k--){
rotateHead=rotateHead->next;
}
ListNode* ans=rotateHead->next;
rotateHead->next=nullptr;
rotateTail->next=head;
return ans;
}
};