难度: m i d d l e \color{orange}{middle} middle
题目描述
给你一个链表的头节点 h e a d head head ,旋转链表,将链表每个节点向右移动 k k k 个位置。
示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]
复制示例输入
示例 2:
输入:head = [0,1,2], k = 4
输出:[2,0,1]
复制示例输入
提示:
- 链表中节点的数目在范围 [ 0 , 500 ] [0, 500] [0,500] 内
- − 100 < = N o d e . v a l < = 100 -100 <= Node.val <= 100 −100<=Node.val<=100
- 0 < = k < = 2 ∗ 1 0 9 0 <= k <= 2 * 10^{9} 0<=k<=2∗109
算法
(模拟,连接) O ( n ) O(n) O(n)
- 这道题中 k 可能很大,所以我们令 k=k%n,n是链表长度。
- 创建两个指针 first, second,分别指向头结点,先让 first 向后移动 k个位置,然后first和second同时向后移动,直到first走到链表最后一个元素。
- 此时first指向链表末尾,second指向分界点。然后我们把链表从分界点处断开,然后把后半段接在前半段前面即可。
C++ 代码
/**
* 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:
ListNode* rotateRight(ListNode* head, int k) {
if(!head) return head;
int n = 0;
ListNode* p = head;
while (p) {
n ++;
p = p->next;
}
k %= n;
if (k == 0) return head;
ListNode* first = head;
while(k -- && first != NULL) first = first->next;
ListNode* second = head;
while (first->next) {
first = first->next;
second = second->next;
}
first->next = head;
head = second->next;
second->next = 0;
return head;
}
};