给定一个链表,旋转链表,将链表每个节点向右移动 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
思路:
首先计算出链表的长度,然后将尾节点指向头结点形成环。找出移动K步后的头结点的前驱节点,具体计算公式是m=len-k%len,将尾节点向前移动m位得到新的头结点的前驱节点。断开链表,返回头结点。画图,带数字进去推导一下即可。
/**
* 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||k==0){
return head;
}
ListNode* tail=head;
int len=1;
while(tail->next){
tail=tail->next;
len++;
}
if(k%len==0){
return head;
}
tail->next=head;//首尾相连,形成环形链表
int m=len-k%len;
while(m--){ //到达新头节点的前驱节点
tail=tail->next;
}
ListNode* res=tail->next;
tail->next=nullptr;
return res;
}
};