Golang程序 在第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