Golang程序 在一个给定的链接列表中添加第一个节点
例子
解决这个问题的方法
第1步 - 定义一个方法,接受链表的头部。
第2步 - 如果 head == nil,创建一个新的节点并返回该节点。
第3步 - 如果 head 不是 nil,则更新输入链表的头部。
例子
package main
import "fmt"
type Node struct {
value int
next *Node
}
func NewNode(value int, next *Node) *Node{
var n Node
n.value = value
n.next = next
return &n
}
func TraverseLinkedList(head *Node){
temp := head
for temp != nil {
fmt.Printf("%d ", temp.value)
temp = temp.next
}
fmt.Println()
}
func AddFirstNode(head *Node, data int) *Node{
if head == nil{
head = NewNode(data, nil)
return head
}
newNode := NewNode(data, nil)
newNode.next = head
return newNode
}
func main(){
head := NewNode(30, NewNode(10, NewNode(40, NewNode(40, nil))))
fmt.Printf("Input Linked list is: ")
TraverseLinkedList(head)
head = AddFirstNode(head, 5)
fmt.Printf("After adding first node, linked list is: ")
TraverseLinkedList(head)
}
输出
Input Linked list is: 30 10 40 40
After adding first node, linked list is: 5 30 10 40 40