给你一个链表的头节点 head
,旋转链表,将链表每个节点向右移动 k
个位置。
分析
记给定链表的长度为 n,注意到当向右移动的次数 k ≥n 时,我们仅需要向右移动 k mod n次即可。因为每 nn次移动都会让链表变为原状。这样我们可以知道,新链表的最后一个节点为原链表的第 (n - 1) - ( k mod n) 个节点(从 0 开始计数)。
将给定的链表链接成环 进行处理
1.首先计算出链表的长度 n,并找到该链表的末尾节点,将其与头节点相连。得到了闭合为环的链表。
2.找到新链表的最后一个节点(即原链表的第 (n - 1) - ( k mod n) 个节点),将当前闭合为环的链表断开,即可得到我们所需要的结果。
3.当链表长度不大于 1,或者 kk为 n 的倍数时,新链表将与原链表相同,我们无需进行任何处理。。
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k)
{
if (k == 0 || head == NULL || head->next == NULL)
{
return head;
}
int n = 1;
ListNode* iter = head;
while (iter->next != NULL) {
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 = NULL;
return ret;
}
};