数据结构与算法【LeetCode-Offer】：2.3链表—707. 设计链表

原文链接

题目描述

题目链接

实例：

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

解题思路

简单题，覆盖了链表的所有基本操作，主要就是链表的长度判断这块需要细心一点：

package com.kami.leetcode.list_study;


import com.kami.leetcode.linked.Node;

import java.util.List;

/**
 * @Description: TODO
 * @author: scott
 * @date: 2021年12月10日 9:48
 * 自定义的单链表结构在下面
 */
public class MyLinkedList {

    //链表元素个数
    private int size;
    //虚拟头节点
    private ListNode1 dummy;


    public MyLinkedList() {
        size = 0;
        dummy = new ListNode1(0);
    }

    /**
     * 获取链表中下标为 index 的节点的值。如果索引无效，则返回-1。
     */
    public int get(int index) {
        if(index < 0 || index >= size){
            return -1;
        }
        ListNode1 cur = dummy;
        // 添加了一个虚拟头节点，所以查找下标 index+1 的节点
        for(int i = 0; i <= index; i++){
            cur = cur.next;
        }

        return cur.val;
    }

    /**
     * 在链表中的第 index 个节点之前添加值为 val  的节点
     * 如果 index 等于链表的长度，则该节点将附加到链表的末尾
     * 如果 index 大于链表长度，则不会插入节点。如果index小于0，则在头部插入节点
     */
    public void addAtIndex(int index, int val) {
        //如果 index 大于链表长度，则不会插入节点
        if(index > size){
            return;
        }
        //如果index小于0，则在头部插入节点
        if(index < 0){
            index = 0;
        }

        size++;

        // 找到要插入节点的前驱
        ListNode1 pre = dummy;
        for(int i = 0; i < index; i++){
            pre = pre.next;
        }

        // 待插入节点
        ListNode1 newNode = new ListNode1(val);
        newNode.next = pre.next;
        pre.next = newNode;
    }

    /**
     * 在链表的第一个元素之前添加一个值为 val 的节点。插入后，新节点将成为链表的第一个节点。
     */
    public void addAtHead(int val) {
        addAtIndex(0, val);
    }

    /**
     * 将值为 val 的节点追加到链表的最后一个元素。
     * @param val
     */
    public void addAtTail(int val) {
        addAtIndex(size, val);
    }


    /**
     * 如果索引 index 有效，则删除链表中的第 index 个节点。
     * @param index
     */
    public void deleteAtIndex(int index) {
        if(index < 0 || index >= size){
            return;
        }

        size--;

        //找到要删除节点的前驱
        ListNode1 pre = dummy;
        for(int i = 0; i < index; i++){
            pre = pre.next;
        }

        //待删除的节点
        ListNode1 delNode = pre.next;
        pre.next = delNode.next;
    }
}

/**
 * 自定义的单链表结构
 */
class ListNode1 {
    int val;
    ListNode1 next;

    ListNode1(int x) {
        this.val = x;
        this.next = null;
    }

    
}

补充了测试类的版本