[路飞]_设计链表

87 阅读2分钟

题目介绍

设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性:val 和 nextval 是当前节点的值,next 是指向下一个节点的指针/引用。如果要使用双向链表,则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是 0-index 的。

在链表类中实现这些功能:

  • get(index):获取链表中第 index 个节点的值。如果索引无效,则返回-1
  • addAtHead(val):在链表的第一个元素之前添加一个值为 val 的节点。插入后,新节点将成为链表的第一个节点。
  • addAtTail(val):将值为 val 的节点追加到链表的最后一个元素。
  • addAtIndex(index,val):在链表中的第 index 个节点之前添加值为 val  的节点。如果 index 等于链表的长度,则该节点将附加到链表的末尾。如果 index 大于链表长度,则不会插入节点。如果index小于0,则在头部插入节点。
  • deleteAtIndex(index):如果索引 index 有效,则删除链表中的第 index 个节点。

示例

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

提示:

解题代码

var Node = function(val, next) {
    this.val = val ? val : 0
    this.next = next ? next : null
}

var MyLinkedList = function() {
    this.head = null
    this.tail = null
    this.size = 0
};

/** 
 * @param {number} index
 * @return {number}
 */
MyLinkedList.prototype.get = function(index) {
    if (index > this.size - 1) return -1
    let p = this.head
    while (index--) {
        p = p.next
    }
    return p.val
};

/** 
 * @param {number} val
 * @return {void}
 */
MyLinkedList.prototype.addAtHead = function(val) {
    if (this.size === 0) {
        this.head = this.tail = new Node(val)
    } else {
        const newHead = new Node(val, this.head)
        this.head = newHead
    }
    this.size++
};

/** 
 * @param {number} val
 * @return {void}
 */
MyLinkedList.prototype.addAtTail = function(val) {
    if (this.size === 0) {
        this.head = this.tail = new Node(val)
    } else {
        this.tail.next = new Node(val)
        this.tail = this.tail.next
    }
    this.size++
};

/** 
 * @param {number} index 
 * @param {number} val
 * @return {void}
 */
MyLinkedList.prototype.addAtIndex = function(index, val) {
    if (index > this.size) return 
    if (index === this.size) {
        this.addAtTail(val)
        return
    }
    if (index === 0) {
        this.addAtHead(val)
        return
    }
    let p = this.head
    while (--index) {
        p = p.next
    }
    const newNode = new Node(val, p.next)
    p.next = newNode
    this.size++
};

/** 
 * @param {number} index
 * @return {void}
 */
MyLinkedList.prototype.deleteAtIndex = function(index) {
    if (index > this.size - 1) return
    if (index === 0) {
        this.head = this.head.next
        if (this.size === 1) this.tail = this.head
    } else {
        let p = this.head, i = index
        while (--index) {
            p = p.next
        }
        p.next = p.next.next
        if (i === this.size - 1) this.tail = p
    }
    this.size--
};

/**
 * Your MyLinkedList object will be instantiated and called as such:
 * var obj = new MyLinkedList()
 * var param_1 = obj.get(index)
 * obj.addAtHead(val)
 * obj.addAtTail(val)
 * obj.addAtIndex(index,val)
 * obj.deleteAtIndex(index)
 */