golang双向链表

发布时间:2024-07-07 17:22:09

双向链表是一种常见的数据结构,它与单向链表相比,每个节点除了存储下一个节点的地址外,还存储前一个节点的地址,这样可以实现双向遍历和操作。在golang中,我们可以通过自定义结构体和指针来实现双向链表。

定义双向链表结构体

Golang是一种静态类型的语言,我们可以通过自定义结构体来定义双向链表的节点。每个节点包含三个字段,分别是前一个节点的指针、当前节点的值,以及后一个节点的指针。

type Node struct {
    prev *Node
    value interface{}
    next *Node
}

初始化双向链表

在使用双向链表之前,我们需要初始化一个空的链表。可以使用一个特殊的头节点和尾节点来代表链表的起始和结束位置,这样在插入和删除节点时会更加方便。

type LinkedList struct {
    head *Node
    tail *Node
}

插入和删除节点

在双向链表中,插入和删除节点是常见的操作。通过操作节点的指针,我们可以在O(1)的时间复杂度内完成这些操作。

插入节点:在双向链表中,插入节点可以分为两种情况,分别是插入到链表的头部和尾部以及其他位置。

对于插入到头部和尾部的情况,我们只需要更新节点的指针即可。例如,当插入节点到头部时:

func (l *LinkedList) InsertToFront(value interface{}) {
    newNode := &Node{
        prev: nil,
        value: value,
        next: l.head,
    }
    if l.head != nil {
        l.head.prev = newNode
    }
    l.head = newNode
    if l.tail == nil {
        l.tail = newNode
    }
}

对于插入到其他位置的情况,我们需要先找到要插入位置节点的前一个节点,并更新指针。例如,当插入节点到索引为index的位置时:

func (l *LinkedList) InsertAtIndex(value interface{}, index int) {
    newNode := &Node{
        prev: nil,
        value: value,
        next: nil,
    }
    if index == 0 {
        newNode.next = l.head
        if l.head != nil {
            l.head.prev = newNode
        }
        l.head = newNode
        if l.tail == nil {
            l.tail = newNode
        }
        return
    }
    prevNode := l.head
    for i := 0; i < index-1; i++ {
        if prevNode == nil {
            return
        }
        prevNode = prevNode.next
    }
    newNode.next = prevNode.next
    newNode.prev = prevNode
    if prevNode.next != nil {
        prevNode.next.prev = newNode
    }
    prevNode.next = newNode
}

删除节点:删除节点与插入节点类似,也需要更新节点的指针。例如,当删除索引为index的节点时:

func (l *LinkedList) DeleteAtIndex(index int) {
    if l.head == nil {
        return
    }
    if index == 0 {
        l.head = l.head.next
        if l.head != nil {
            l.head.prev = nil
        }
        if l.head == nil {
            l.tail = nil
        }
        return
    }
    prevNode := l.head
    for i := 0; i < index-1; i++ {
        if prevNode == nil {
            return
        }
        prevNode = prevNode.next
    }
    if prevNode.next == nil {
        return
    }
    if prevNode.next == l.tail {
        l.tail = prevNode
    }
    prevNode.next = prevNode.next.next
    if prevNode.next != nil {
        prevNode.next.prev = prevNode
    }
}

遍历和访问节点

使用双向链表,我们可以很方便地遍历和访问节点。

func (l *LinkedList) Traverse() {
    currNode := l.head
    for currNode != nil {
        fmt.Println(currNode.value)
        currNode = currNode.next
    }
}

func (l *LinkedList) Get(index int) interface{} {
    currNode := l.head
    for i := 0; i < index; i++ {
        if currNode == nil {
            return nil
        }
        currNode = currNode.next
    }
    if currNode == nil {
        return nil
    }
    return currNode.value
}

通过遍历节点,我们可以将链表中的所有元素打印出来或者对每个节点进行操作。通过访问节点,我们可以根据索引获取对应的值。

使用上述方式,我们就可以在golang中实现一个双向链表的数据结构了。双向链表在某些场景下可以提供更高效的操作和便利的特性,比如在需要经常进行插入和删除操作的情况下。希望这篇文章对你理解和使用双向链表有所帮助!

相关推荐