[leetCode]旋轉鏈表的k個節點

旋轉鏈表的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;
    }
};

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章