Go语言程序:在第K个节点之后插入新节点。
示例
在值为10的节点之后添加节点15。
解决这个问题的方法
步骤1 − 定义一个接受链表头的方法。
步骤2 − 如果head == nil,则返回head。
步骤3 − 遍历给定的链表。
步骤4 − 如果temp.value为10,则将节点15添加为下一个节点。
步骤5 − 如果找不到值为10的节点,则不添加任何节点并返回head。
示例
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 AddAfterKthNode(head *Node, k , data int) *Node{ // Insert node after Kth node. if head == nil{ return head } temp := head for temp != nil{ if temp.value == k{ newNode := NewNode(data, nil) newNode.next = temp.next temp.next = newNode break } temp = temp.next } return head } func main(){ head := NewNode(30, NewNode(10, NewNode(40, NewNode(40, nil)))) fmt.Printf("Input Linked list is: ") TraverseLinkedList(head) head = AddAfterKthNode(head, 10, 15) fmt.Printf("Added node after %dth value node, Linked List is: ", 10) TraverseLinkedList(head) }
输出
Input Linked list is: 30 10 40 40 Added node after 10th value node, Linked List is: 30 10 15 40 40
广告