给定一个链表,旋转链表,将链表每个节点向右移动 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
思路:将链表中的数据导入到一个容器中,然后对容器中的数据进行旋转,旋转完成后,将原链表中每一个节点的val修改为与容器相同即可
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(head==NULL || head->next==NULL){
return head;
}
vector<int> a;
int length=0;
ListNode* p=head;
while(p!=NULL){
length++;//求出链表长度
a.push_back(p->val);//返回到数组中
p=p->next;
}
k=k%length;//因为实际上进行length次旋转就是没有旋转
while(k>0){
auto it1=a.insert(a.begin(),a[a.size()-1]);//将最后一个数字移动到第一个
a.pop_back();//删除最后一个
k--;
}
ListNode* q=head;
for(int i=0;i<length;i++){//将head中每个位置的数值改变为与a容器中的一致
q->val=a[i];
q=q->next;
}
return head;
}
};