【leetcode面试经典150题】专栏系列将为准备暑期实习生以及秋招的同学们提高在面试时的经典面试算法题的思路和想法。本专栏将以一题多解和精简算法思路为主,题解使用C++语言。(若有使用其他语言的同学也可了解题解思路,本质上语法内容一致)
【题目描述】
给你一个链表的头节点 head
,旋转链表,将链表每个节点向右移动 k
个位置。
【示例一】
输入:head = [1,2,3,4,5], k = 2 输出:[4,5,1,2,3]
【示例二】
输入:head = [0,1,2], k = 4 输出:[2,0,1]
【提示及数据范围】
- 链表中节点的数目在范围
[0, 500]
内 -100 <= Node.val <= 100
0 <= k <= 2 * 10的9次方
【代码】
// 闭合为环
// 记给定链表的长度为 n,注意到当向右移动的次数 k≥n 时,
// 我们仅需要向右移动 k mod n次即可。因为每 n 次移动都会让链表变为原状。
// 这样我们可以知道,新链表的最后一个节点为原链表的第 (n−1)−(k mod n) 个节点(从 0 开始计数)。
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if (k == 0 || head == nullptr || head->next == nullptr) {
return head;
}
int n = 1;
ListNode* iter = head;
while (iter->next != nullptr) {
iter = iter->next;
n++;
}
int add = n - k % n;
if (add == n) {
return head;
}
iter->next = head;
while (add--) {
iter = iter->next;
}
ListNode* ret = iter->next;
iter->next = nullptr;
return ret;
}
};