设计链表
题目
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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))
}