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
}