题目:
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]
输入:head = [0,1,2], k = 4
输出:[2,0,1]
思路:
通过数学观察发现,旋转链表实际上就是把倒数第k%n个结点及其以后的结点提前,然后在其末尾连上前面的结点。但是要注意如果k%n等于0则可以预先处理(即无移动),否则还要修改相应的代码。
代码:
/**
* 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:
//取模,第倒数k%n个以后的链表提前,之前的链表再连上去,即为旋转链表
int getLength(ListNode* head){
int len=0;
while(head){
len++;
head=head->next;
}
return len;
}
ListNode* rotateRight(ListNode* head, int k) {
if(!head)
return head;
ListNode* head1,*head2,*prev,*curr;
int n=getLength(head);
int index=k%n;
head1=new ListNode();
head2=new ListNode();
head1->next=head;
prev=head1,curr=head;
if(n==1||index==0)
return head;
for(int i=1;i<n-index+1;i++){
prev=prev->next;
curr=curr->next;
}
prev->next=NULL;
head2->next=curr;
while(curr->next){
curr=curr->next;
}
curr->next=head1->next;
return head2->next;
}
};