61. Rotate List
61. Rotate List
题目
Given a list, rotate the list to the right by k places, where k is non-negative.
Example:
Given 1->2->3->4->5->NULL and k = 2,
return 4->5->1->2->3->NULL.
解析
题目;给定一个链表,将链表旋转到右边的k个位置,其中k是非负的。
例如:
1->2->3->4->5->NULL,为K = 2,
返还4->5->1->2->3->NULL。
/
/
分析:先遍历一遍,得出链表长度len,注意k可能会大于len,因此k%=len。
将尾结点next指针指向首节点,形成一个环,接着往后跑len-k步,从这里断开,就是结果
*/
class Solution_61 {
public:
//input:[1, 2] 3
//output:[2, 1]
ListNode *rotateRight(ListNode *head, int k) {
if (!head||k==0||!head->next)
{
return head;
}
ListNode* newHead = head;
ListNode* fast = head;
ListNode* lastNode = NULL;
int len = 0;
while (fast!=nullptr)
{
if (fast->next==NULL)
{
lastNode = fast;
}
fast = fast->next;
len++;
}
fast = head;
int step = len - k%len-1; // 注意k可能会大于len,因此k%=len
if (k%len == 0)
{
return newHead;
}
while (step)
{
step--;
fast = fast->next;
}
newHead = fast->next;
fast->next = NULL;
lastNode->next = head;
return newHead;
}
};
链接:https://www.nowcoder.com/questionTerminal/afbec6b9d4564c63b2c149ccc01c5678
来源:牛客网
class Solution {
public:
ListNode *rotateRight(ListNode *head, int k) {
if(head==nullptr||k==0)
return head;
int len=1;
ListNode *p=head;
while(p->next){
//遍历一遍,求出链表长度
len++;
p=p->next;
}
k=len-k%len;
p->next=head;//首尾相连
for(int step=0;step<k;step++){
p=p->next;//接着往后跑
}
head=p->next;//新的首节点
p->next=nullptr;//断开环
return head;
}
};
题目来源
C/C++基本语法学习
STL
C++ primer