题目描述
Given the head of a linked list, rotate the list to the right by k places.
解答
/**
* 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(k=0 || head == nullptr || head->next == nullptr) return head;
ListNode * temp = head;
ListNode * pre = temp;
uint32_t flag = 1;
while(temp->next != nullptr)
{
flag++;
temp = temp->next;
}
k = k % flag;
temp = head;
for(int i = 0; i < k; ++i)
{
while(temp->next != nullptr)
{
pre = temp;
temp = temp->next;
}
pre->next = nullptr;
temp->next = head;
head = temp;
}
return head;
}
};
思路
初始思路
移动的次数,就等于尾节点变成头节点的次数.但是这会遇到一个问题,就是移动的次数大于链表的长度.比如说链表的长度是3,但是却要移动2000000次.如果还这样逐次移动很容易超出时间限制.
改进方法
对移动次数k进行取余运算,即k = k % flag
,这样就把移动的次数限制在链表的长度范围内,大大降低了时间的消耗.
边界问题
k=0
、链表长度为零或一都直接返回原链表.
更好的方法
参考官方题解