2021-03-27:給你一個鏈表的頭節點 head ,旋轉鏈表,將鏈表每一個節點向右移動 k 個位置

2021-03-27:給你一個鏈表的頭節點 head ,旋轉鏈表,將鏈表每一個節點向右移動 k 個位置。輸入:head = 1→2→3→4→5, k = 2,輸出:4→5→1→2→3。golang

福大大 答案2020-03-27:緩存

1.找尾節點而且計算鏈表節點個數。
2.若是k大於等鏈表節點個數,須要取模,k必定在[0,節點個數)範圍以內。若是k=0,直接返回頭節點。
3.求倒數k+1的節點。
4.緩存倒數第k節點ans。
5.尾節點連頭節點。
6.倒數k+1節點的Next指針爲空。
7.返回ans。ide

代碼用golang編寫,代碼以下:指針

package main

import "fmt"

func main() {
    head := &ListNode{Val: 1}
    head.Next = &ListNode{Val: 2}
    head.Next.Next = &ListNode{Val: 3}
    head.Next.Next.Next = &ListNode{Val: 4}
    printlnLinkNodeList(head)
    k := 6
    fmt.Println("k =", k)
    fmt.Println("----------------")
    ret := rotateRight(head, k)
    printlnLinkNodeList(ret)
}

type ListNode struct {
    Val  int
    Next *ListNode
}

func rotateRight(head *ListNode, k int) *ListNode {
    if head == nil {
        return head
    }
    //找尾節點而且計數
    cnt := 1
    tail := head
    for tail.Next != nil {
        cnt++
        tail = tail.Next
    }

    k = k % cnt
    if k == 0 { //恰好是頭節點,就不用操做了。
        return head
    }

    //找倒數第k+1節點
    fast := head
    slow := head
    k++
    for k > 0 {
        fast = fast.Next
        k--
    }
    for fast != nil {
        fast = fast.Next
        slow = slow.Next
    }

    //緩存結果
    ans := slow.Next
    //尾節點連頭節點
    tail.Next = head
    //倒數k+1節點無Next指針
    slow.Next = nil

    return ans
}

//鏈表打印
func printlnLinkNodeList(head *ListNode) {
    cur := head
    for cur != nil {
        fmt.Print(cur.Val, "\t")
        cur = cur.Next
    }
    fmt.Println()
}

執行結果以下:
在這裏插入圖片描述code


力扣61. 旋轉鏈表
評論blog

相關文章
相關標籤/搜索