1. 程式人生 > >LeetCode-61-Rotate List

LeetCode-61-Rotate List

span nbsp pre steps amp style count out ace

算法描述:

Given a linked list, rotate the list to the right by k places, where k is non-negative.

Example 1:

Input: 1->2->3->4->5->NULL, k = 2
Output: 4->5->1->2->3->NULL
Explanation:
rotate 1 steps to the right: 5->1->2->3->4->NULL
rotate 2 steps to the right: 4->5->1->2->3->NULL

Example 2:

Input: 0->1->2->NULL, k = 4
Output: 2->0->1->NULL
Explanation:
rotate 1 steps to the right: 2->0->1->NULL
rotate 2 steps to the right: 1->2->0->NULL
rotate 3 steps to the right: 0->1->2->NULL
rotate 4 steps to the right: 2->0->1->NULL

解題思路:單鏈表的題目,畫出圖就可以解決了。註意邊界條件,以及k值大於鏈表長度等問題。

    ListNode* rotateRight(ListNode* head, int k) {
        if(head==nullptr || k ==0) return head;
        ListNode* dup = new ListNode(-1);
        dup->next = head;
        ListNode* fast = dup;
        int count =0;
        while(fast->next!=nullptr){
            fast=fast->next;
            count
++; } int n = count - k%count; ListNode* slow = dup; while(n>0){ slow = slow->next; n--; } fast->next = dup->next; dup->next = slow->next; slow->next = nullptr; return dup->next; }

LeetCode-61-Rotate List