在Go语言中实现双向链表时,AddHead方法用于向链表头部插入新节点,若没有正确处理不同链表状态下的指针关系,很容易触发nil指针恐慌。双向链表每个节点需要同时持有前驱和后继节点的指针,头节点的前驱指针通常为nil,尾节点的后继指针通常为nil。

双向链表的基础结构定义
首先我们需要定义双向链表的节点和链表结构体,节点需要包含存储的值、指向前驱节点的指针和指向后继节点的指针,链表结构体只需要持有头节点的指针即可,若需要快速访问尾节点也可以额外添加尾节点指针,本文先实现基础仅含头指针的版本。
// 双向链表节点定义
type Node struct {
Val interface{} // 节点存储的值
Prev *Node // 前驱节点指针
Next *Node // 后继节点指针
}
// 双向链表定义
type DoubleLinkedList struct {
Head *Node // 头节点指针
}
AddHead方法的逻辑分析
AddHead方法的核心逻辑需要分两种场景处理,分别是当前链表为空(Head为nil)和当前链表不为空(Head不为nil)的情况,两种场景的指针操作完全不同,遗漏任何一种都会导致指针恐慌。
场景一:链表为空
当链表为空时,新插入的节点既是头节点也是尾节点,此时它的Prev和Next指针都应该设置为nil,然后将链表的Head指针指向这个新节点即可,不需要操作其他节点的指针,因为此时没有其他节点存在。
场景二:链表不为空
当链表已经存在节点时,新节点需要成为新的头节点,原有头节点需要变为新节点的后继节点。此时需要先让新节点的Next指向原有头节点,再让原有头节点的Prev指向新节点,最后将链表的Head指针更新为新节点,注意操作顺序不能颠倒,否则会丢失原有头节点的引用。
完整AddHead方法实现
结合上述两种场景,我们可以写出完整的AddHead方法实现,代码如下:
// AddHead 向双向链表头部插入新节点
func (l *DoubleLinkedList) AddHead(val interface{}) {
// 创建新节点,初始前驱和后继都为nil
newNode := &Node{
Val: val,
Prev: nil,
Next: nil,
}
// 判断链表是否为空
if l.Head == nil {
// 空链表场景,新节点直接作为头节点
l.Head = newNode
} else {
// 非空链表场景,处理指针关系
newNode.Next = l.Head // 新节点的后继指向原有头节点
l.Head.Prev = newNode // 原有头节点的前驱指向新节点
l.Head = newNode // 更新链表头节点为新节点
}
}
边界情况测试验证
为了验证AddHead方法不会出现nil指针恐慌,我们可以编写测试用例覆盖不同场景:
package main
import "fmt"
func main() {
// 测试1:空链表插入第一个节点
list1 := &DoubleLinkedList{}
list1.AddHead(1)
fmt.Println("空链表插入后头节点值:", list1.Head.Val) // 输出1
fmt.Println("头节点前驱是否为nil:", list1.Head.Prev == nil) // 输出true
fmt.Println("头节点后继是否为nil:", list1.Head.Next == nil) // 输出true
// 测试2:非空链表插入第二个节点
list1.AddHead(2)
fmt.Println("插入后头节点值:", list1.Head.Val) // 输出2
fmt.Println("新头节点后继值:", list1.Head.Next.Val) // 输出1
fmt.Println("原头节点前驱值:", list1.Head.Next.Prev.Val) // 输出2
// 测试3:连续插入多个节点
list1.AddHead(3)
list1.AddHead(4)
cur := list1.Head
for cur != nil {
fmt.Print(cur.Val, " ") // 输出4 3 2 1
cur = cur.Next
}
}
常见错误规避
很多开发者实现AddHead方法时容易犯两个错误:一是在非空链表场景下先更新l.Head为新节点,再操作原有头节点的指针,此时原有头节点的引用已经丢失,访问l.Head.Prev会操作新节点,导致指针关系错误;二是忘记处理空链表场景,直接按照非空链表的指针操作逻辑执行,此时l.Head为nil,访问l.Head.Prev就会触发nil指针恐慌。按照本文的分场景处理逻辑,就可以完全避免这类问题。