旋轉鏈表的k個節點可以使用快慢指針來解決。
測試用例當時沒有跑過,後來仔細想想,沒有考慮到一點就是參數k在後臺可能是給的很大,遠遠大於count的時候,整個鏈表就會旋轉的特別亂,所以需要將k控制到count的範圍內。
在線OJ的代碼如下:
class Solution {
public:
ListNode *rotateRight(ListNode *head, int k) {
int count = 0;
//判斷是否爲空
if(head==NULL)
{
return NULL;
}
ListNode* node = head;
//統計該鏈表有多少個節點
while(node)
{
count++;
node = node->next;
}
k %= count;
ListNode *fast = head, *slow = head;
for (int i = 0; i < k; ++i) {
if (fast)
fast = fast->next;
}
if (!fast)
return head;
while (fast->next) {
fast = fast->next;
slow = slow->next;
}
fast->next = head;
fast = slow->next;
slow->next = NULL;
return fast;
}
};