Medium!
題目描述:
給定一個鏈表,旋轉鏈表,將鏈表每個節點向右移動 k 個位置,其中 k 是非負數。
示例 1:
輸入: 1->2->3->4->5->NULL, k = 2 輸出: 4->5->1->2->3->NULL 解釋: 向右旋轉 1 步: 5->1->2->3->4->NULL 向右旋轉 2 步: 4->5->1->2->3->NULL
示例 2:
輸入: 0->1->2->NULL, k = 4 輸出:2->0->1->NULL解釋: 向右旋轉 1 步: 2->0->1->NULL 向右旋轉 2 步: 1->2->0->NULL 向右旋轉 3 步:0->1->2->NULL向右旋轉 4 步:2->0->1->NULL
解題思路:
這道旋轉鏈表的題和之前那道 Rotate Array 旋轉數組 很類似,但是比那道要難一些,因為鏈表的值不能通過下標來訪問,只能一個一個的走,我們可能首先想到的就是用快慢指針來解,快指針先走k步,然后兩個指針一起走,當快指針走到末尾時,慢指針的下一個位置是新的順序的頭結點,這樣就可以旋轉鏈表了,自信滿滿地寫完程序,放到OJ上跑,以為能一次通過,結果就跪在了各種特殊情況,首先一個就是當原鏈表為空時,直接返回NULL,還有就是當k大於鏈表長度和k遠遠大於鏈表長度時該如何處理,我們需要首先遍歷一遍原鏈表得到鏈表長度n,然后k對n取余,這樣k肯定小於n,就可以用上面的算法了。
C++解法一:
1 class Solution { 2 public: 3 ListNode *rotateRight(ListNode *head, int k) { 4 if (!head) return NULL; 5 int n = 0; 6 ListNode *cur = head; 7 while (cur) { 8 ++n; 9 cur = cur->next; 10 } 11 k %= n; 12 ListNode *fast = head, *slow = head; 13 for (int i = 0; i < k; ++i) { 14 if (fast) fast = fast->next; 15 } 16 if (!fast) return head; 17 while (fast->next) { 18 fast = fast->next; 19 slow = slow->next; 20 } 21 fast->next = head; 22 fast = slow->next; 23 slow->next = NULL; 24 return fast; 25 } 26 };
這道題還有一種解法,跟上面的方法類似,但是不用快慢指針,一個指針就夠了,原理是先遍歷整個鏈表獲得鏈表長度n,然后此時把鏈表頭和尾鏈接起來,再往后走n - k % n個節點就到達新鏈表的頭結點前一個點,這時斷開鏈表即可。
C++解法二:
1 class Solution { 2 public: 3 ListNode *rotateRight(ListNode *head, int k) { 4 if (!head) return NULL; 5 int n = 1; 6 ListNode *cur = head; 7 while (cur->next) { 8 ++n; 9 cur = cur->next; 10 } 11 cur->next = head; 12 int m = n - k % n; 13 for (int i = 0; i < m; ++i) { 14 cur = cur->next; 15 } 16 ListNode *newhead = cur->next; 17 cur->next = NULL; 18 return newhead; 19 } 20 };
