将数组转换为Golang的链接列表

Question

链接列表的定义：

type ListNode struct {
    Val  int
    Next *ListNode
}

插入指针操作的助手： 我知道root.val == 0不是解决输入数组包含0个元素的问题，因此请提出一种更通用的方法来解决此问题。

func insert(root *ListNode, elem int) *ListNode {
    temp := ListNode{Val: elem, Next: nil}
    if root.Val == 0 {
        root = &temp
    } else {
        curr := root
        for curr.Next != nil {
            curr = curr.Next
        }
        curr = &temp
    }
    return root
}

主要功能：

func convertToList(arr []int) *ListNode {
    var head ListNode
    for _, val := range arr {
        head = *insert(&head, val)
    }
    return &head
}

测试功能的字符串函数实现：

func (l *ListNode) String() string {
    x := make([]int, 0)
    curr := l
    for curr != nil {
        x = append(x, curr.Val)
        curr = curr.Next
    }
    return fmt.Sprint(x)

}

我要复制输出的主要功能：

func main() {
    arr := []int{1, 2, 3, 4, 5}
    listNode := convertToList(arr)
    fmt.Println(listNode.String())
}

输出：

[1]

预期输出：

[1 2 3 4 5]

Answer 1

您必须更改converttolist和insert函数为：

func insert(root *ListNode, elem int) *ListNode {
    temp := ListNode{Val: elem, Next: nil}
    if root == nil {
        root = &temp
        return root
    }
    curr := root
    for curr.Next != nil {
        curr = curr.Next
    }
    curr.Next = &temp
    return root
}

和converttolist：：

func convertToList(arr []int) *ListNode {
    var head *ListNode
    for _, val := range arr {
        head = insert(head, val)
    }
    return head
}

Answer 2

更改converttolist：

func convertToList(arr []int) *ListNode {
    head := new(ListNode)
    if len(arr) == 0 {
        return head
    }
    head.Val = arr[0]
    head.Next = nil

    last := head
    for i := 1; i < len(arr); i++ {
        last = insert(last, arr[i])
    }
    return head
}

和insert：

func insert(last *ListNode, val int) *ListNode {
    temp := new(ListNode)
    temp.Val = val
    temp.Next = nil
    last.Next = temp
    return last.Next
}

请参阅此类似：链接列表有关更多信息