问题:
给你一个链表的头节点 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]
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(head == nullptr)return nullptr;
ListNode *p = head;
//计算链表长度
int cnt = 1;
//将单链表首尾相连
while(p->next){
p = p->next,cnt++;
}
p->next = head;
//计算偏移量
k %= cnt,k = cnt-k;
//将偏移量后面命为head指针,并断开链表
while(k--){
p = p->next;
}
head = p->next;
p->next = nullptr;
//返回偏移后的链表
return head;
}
};