leetcode 探索 鏈表 旋轉鏈表

題目

給定一個鏈表，旋轉鏈表，將鏈表每個節點向右移動 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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/rotate-list

分析

說是旋轉鏈表，其實就是一個將尾部倒數第K個元素，作爲新的頭，將尾部指向原先的頭。但是要注意K是有可能大於鏈表的Size的，所以其實可以先遍歷一遍獲取size，然後K % Size取餘，然後在前進頭指針，獲取Size-K的指針，就是新的頭了。尾指針其實可以在遍歷獲取Size的時候，先取出來。

以下是我一開始的解法，僅做個參考，我是遍歷加獲取元素一起做的，稍微複雜了點。

解法

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func rotateRight(head *ListNode, k int) *ListNode {
    if head == nil || k == 0 {
        return head
    }
    
    // 先前進K步，前進的同時，將k取餘
    cur := head
    var count int
    for i:= 0; i < k;  {
        count++
        cur = cur.Next
        // k 要是大於鏈表的長度
        if cur == nil {
            k = k % count
            count = 0
            i = 0
            cur = head
            // 如果是0，那麼就不需要再輪詢一遍
            if k == 0 {
                break
            }
        } else {
            i++
        }
    }
    // fmt.Printf("%v, size: %d", cur, size)
    // 若是返回了走了K步到了，head上，那麼直接返回就好了。
    if cur == head {
        return head
    }
    
    // 然後快慢指針，找到距離末尾K步的指針，這將是新的頭
    slow, fast := head, cur
    for fast.Next != nil  {
        slow = slow.Next
        fast = fast.Next
    }
    // fmt.Printf("%v", slow)
    
    // 首尾相連，並斷開原來的連接。
    mHead := slow.Next
    slow.Next = nil
    fast.Next = head
    
    return mHead
}