mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2024-09-02 14:20:01 +00:00
44 lines
1.5 KiB
Markdown
44 lines
1.5 KiB
Markdown
# [61. Rotate List](https://leetcode.com/problems/rotate-list/)
|
||
# 思路
|
||
题目要求循环移位链表k步。
|
||
不要被题目迷惑了,这题根本不需要移位,只需要在倒数第k个位置处将链表截断然后将后半部分接到最前面就可以了。现在问题就是怎样快速找到倒数第k个的位置。
|
||
涉及到链表倒数某个位置的一般就是用一个双指针pre和p,初始都为head,先让p后移k步再让pre和p同时后移,这样当p到达链尾时pre就处于倒数第k个位置。
|
||
注意k可能大于链表的长度,所以要先将k对链表长度len取模。
|
||
时间复杂度O(n),空间复杂度O(1)
|
||
|
||
# C++
|
||
``` C++
|
||
/**
|
||
* Definition for singly-linked list.
|
||
* struct ListNode {
|
||
* int val;
|
||
* ListNode *next;
|
||
* ListNode(int x) : val(x), next(NULL) {}
|
||
* };
|
||
*/
|
||
class Solution {
|
||
public:
|
||
ListNode* rotateRight(ListNode* head, int k) {
|
||
if(!head) return head;
|
||
int len = 0;
|
||
ListNode *p = head, *pre=head, *res;
|
||
while(p){
|
||
len++;
|
||
p = p -> next;
|
||
} // 获得链表长度
|
||
k %= len;
|
||
if(k == 0) return head;
|
||
p = head;
|
||
while(k--) p = p -> next; // p先后移k步
|
||
while(p -> next){ // pre和p同时后移直到p -> next为NULL
|
||
p = p -> next;
|
||
pre = pre -> next;
|
||
} // 此时pre -> next 就是倒数第k个元素
|
||
res = pre -> next;
|
||
pre -> next = NULL;
|
||
p -> next = head;
|
||
return res;
|
||
}
|
||
};
|
||
```
|