算法题
链表(linked list)
设计链表

设计链表

题目

你可以选择使用单链表或者双链表,设计并实现自己的链表。

单链表中的节点应该具备两个属性:val 和 next 。val 是当前节点的值,next 是指向下一个节点的指针/引用。

如果是双向链表,则还需要属性 prev 以指示链表中的上一个节点。假设链表中的所有节点下标从 0 开始。

实现 MyLinkedList 类:

  • MyLinkedList() 初始化 MyLinkedList 对象。
  • int get(int index) 获取链表中下标为 index 的节点的值。如果下标无效,则返回 -1 。
  • void addAtHead(int val) 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后,新节点会成为链表的第一个节点。
  • void addAtTail(int val) 将一个值为 val 的节点追加到链表中作为链表的最后一个元素。
  • void addAtIndex(int index, int val) 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。如果 index 等于链表的长度,那么该节点会被追加到链表的末尾。如果 index 比长度更大,该节点将 不会插入 到链表中。
  • void deleteAtIndex(int index) 如果下标有效,则删除链表中下标为 index 的节点。

示例:

输入
["MyLinkedList", "addAtHead", "addAtTail", "addAtIndex", "get", "deleteAtIndex", "get"]
[[], [1], [3], [1, 2], [1], [1], [1]]

输出
[null, null, null, null, 2, null, 3]

解释
MyLinkedList myLinkedList = new MyLinkedList();
myLinkedList.addAtHead(1);
myLinkedList.addAtTail(3);
myLinkedList.addAtIndex(1, 2);    // 链表变为 1->2->3
myLinkedList.get(1);              // 返回 2
myLinkedList.deleteAtIndex(1);    // 现在,链表变为 1->3
myLinkedList.get(1);              // 返回 3

提示:

0 <= index, val <= 1000
请不要使用内置的 LinkedList 库。
调用 get、addAtHead、addAtTail、addAtIndex 和 deleteAtIndex 的次数不超过 2000 。

思路

这道题目设计链表的五个接口:

  • 获取链表第index个节点的数值
  • 在链表的最前面插入一个节点
  • 在链表的最后面插入一个节点
  • 在链表第index个节点前面插入一个节点
  • 删除链表的第index个节点

可以说这五个接口,已经覆盖了链表的常见操作,是练习链表操作非常好的一道题目

链表操作的两种方式:

  • 直接使用原来的链表来进行操作。
  • 设置一个虚拟头结点在进行操作。

解法

下面使用设置一个虚拟头结点的方法:

// 单链表实现
 
type ListNode struct {
    Val  int        // 节点的值
    Next *ListNode  // 下一个节点的指针
}
 
type MyLinkedList struct {
    Head *ListNode // 虚拟头结点
    Size int       // 链表大小
}
 
func Constructor() MyLinkedList {
    return MyLinkedList{Head: &ListNode{}, Size: 0}
}
 
// 获取链表中下标为 index 的节点的值
func (this *MyLinkedList) Get(index int) int {
    if this == nil || index < 0 || index >= this.Size {
        return -1
    }
    cur := this.Head
    for i := 0; i <= index; i++ {
        cur = cur.Next
    }
    return cur.Val
}
 
// 将一个值为 val 的节点插入到链表中第一个元素之前
func (this *MyLinkedList) AddAtHead(val int) {
    this.AddAtIndex(0, val)
}
 
// 将一个值为 val 的节点追加到链表中作为链表的最后一个元素
func (this *MyLinkedList) AddAtTail(val int) {
    this.AddAtIndex(this.Size, val)
}
 
// 将一个值为 val 的节点插入到链表中下标为 index 的节点之前
func (this *MyLinkedList) AddAtIndex(index int, val int) {
    if index < 0 || index > this.Size {
        return
    }
    cur := this.Head
    for i := 0; i < index; i++ {
        cur = cur.Next
    }
    cur.Next = &ListNode{Val: val, Next: cur.Next}
    this.Size++
}
 
// 如果下标有效,则删除链表中下标为 index 的节点
func (this *MyLinkedList) DeleteAtIndex(index int) {
    if index < 0 || index >= this.Size {
        return
    }
    cur := this.Head
    for i := 0; i < index; i++ {
        cur = cur.Next
    }
    if cur.Next != nil {
        cur.Next = cur.Next.Next
        this.Size--
    }
}
 
func (this *MyLinkedList) Print() {
    cur := this.Head.Next
    for cur != nil {
        fmt.Println(cur.Val)
        cur = cur.Next
    }
}
 
func main() {
    obj := Constructor()
    obj.AddAtHead(1)
    obj.AddAtTail(3)
    obj.AddAtIndex(1, 2)
    fmt.Println(obj.Get(1))
    obj.DeleteAtIndex(1)
    fmt.Println(obj.Get(1))
}