自定义ArrayList

语言学习

自定义ArrayList

/**
 * @Author:meihongsheng
 * @Description:com.mhs.test03
 */
public class TestArrayList03<E>{
    /**
     * 设置起始数组
     */
    private Object[] elementData;
    private int size;
    /**
     * 设置起始容量为10
     */
    private static final int DEFAULT_CAPACITY = 10;

    /**
     * 无参构造
     */
    public TestArrayList03(){
        this.elementData = new Object[DEFAULT_CAPACITY];
    }

    /**
     * 根据传入参数指定初始化数组长度
     * @param capacity
     */
    public TestArrayList03(int capacity){
        if(capacity < 0){
            throw new RuntimeException("数组索引不能为负数");
        }else if(capacity == 0){
            this.elementData = new Object[DEFAULT_CAPACITY];
        }else{
            this.elementData = new Object[capacity];
        }
    }

    /**
     * 重写toString方法
     * @return
     */
    @Override
    public String toString() {
        StringBuilder str = new StringBuilder();
        str.append("[");
        for (int i = 0; i < size; i++) {
            if(size - 1 == i){
                str.append(this.elementData[i] + "]");
            }else{
                str.append(this.elementData[i] + ", ");
            }
        }
        return  str.toString();
    }

    /**
     * 添加数组元素的方法
     * @param e
     */
    public void add(E e){
        if(size == elementData.length){
            //扩容
            Object[] newArray = new Object[elementData.length + (elementData.length >> 1)];
            System.arraycopy(elementData,0,newArray,0,elementData.length);
            elementData = newArray;
        }
        //后++解决了数组索引的起始位置问题
        this.elementData[size++] = e;
    }

    /**
     * 返回指定索引的对象
     * @param index
     * @return
     */
    public E get(int index){
        return (E) elementData[index];
    }

    /**
     * 设定指定索引处的对象
     * @param e
     * @param index
     */
    public void set(E e,int index){
        checkIndex(index);
        elementData[index] = e;
    }

    /**
     * 对索引值进行判定
     * @param index
     */
    public void checkIndex(int index){
        if(index < 0 || index > size - 1){
            throw new IndexOutOfBoundsException("索引超出范围" );
        }
    }

    /**
     * 移除指定索引位置的元素
     * @param index
     * @return
     */
    public E remove(int index){
        int len = elementData.length - index - 1;
        checkIndex(index);
        E e  = (E) elementData[index];
        Object[] newArray = elementData;
        System.arraycopy(elementData,index + 1,newArray,index,len);
        newArray[elementData.length - 1] = null;
        this.elementData = newArray;
        size--;
        return e;
    }

    /**
     * 根据指定元素移除
     * @param e
     * @return
     */
    public boolean remove(E e){
        for (int i = 0; i < size; i++) {
            if(elementData[i].equals(e)){
                remove(i);
                return true;
            }
        }
        return false;
    }

    public static void main(String[] args) {
        TestArrayList03<String> tArray = new TestArrayList03<>();
        for (int i = 0; i < 10; i++) {
            tArray.add("梅" + i);
        }
        System.out.println(tArray);
        /*tArray.set("洪",9);
        System.out.println(tArray.get(9));*/
        tArray.remove("梅8");
        System.out.println(tArray.toString());
    }
}