linear-list/linked-list/rotate-list
优质
小牛编辑
128浏览
2023-12-01
Rotate List
描述
Given a list, rotate the list to the right by k
places, where k
is non-negative.
For example: Given 1->2->3->4->5->nullptr
and k = 2
, return 4->5->1->2->3->nullptr
.
分析
先遍历一遍,得出链表长度len
,注意k
可能大于len
,因此令k %= len
。将尾节点 next 指针指向首节点,形成一个环,接着往后跑len-k
步,从这里断开,就是要求的结果了。
代码
// Remove Rotate List
// 时间复杂度O(n),空间复杂度O(1)
class Solution {
public ListNode rotateRight(ListNode head, int k) {
if (head == null || k == 0) return head;
int len = 1;
ListNode p = head;
while (p.next != null) { // 求长度
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 = null; // 断开环
return head;
}
};
// Remove Rotate List
// 时间复杂度O(n),空间复杂度O(1)
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 != nullptr) { // 求长度
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;
}
};