Java 实现简单的自定义ArrayList

public class MyArrayList<E> {

    private Object[] elementData;
    private int size;
    private static final int DEFAULT_CAPACITY = 10;

    public MyArrayList() {
        this.elementData = new Object[DEFAULT_CAPACITY];
    }

    public MyArrayList(int capacity) {
        if (capacity < 0) {
            throw new RuntimeException("容器的容量不能为负数");
        } else if (capacity == 0) {
            this.elementData = new Object[DEFAULT_CAPACITY];
        } else {
            elementData = new Object[capacity];
        }
    }

    public void add(E element) {
        //什么时候扩容
        if (size == elementData.length) {
            //怎么扩容
            Object[] newArray = new Object[elementData.length + (elementData.length >> 1)];//10-->10+10/2=15
            System.arraycopy(elementData, 0, newArray, 0, elementData.length);
            elementData = newArray;
        }
        elementData[size++] = element;
    }

    public E get(int index) {
        checkRange(index);
        return (E) elementData[index];
    }

    public void set(int index, E element) {
        checkRange(index);
        elementData[index] = element;
    }

    public int size(){
        return size;
    }

    public boolean isEmpty(){
        return size==0;
    }

    public void checkRange(int index) {
        //索引合法判断[0,size] 10 0~9
        if (index < 0 || index > size - 1) {
            //不合法
            throw new RuntimeException("索引不合法" + index);
        }
    }

    public void remove(E element) {
        //element,将它和所有元素挨个比较，获得第一个比较为true的，返回。
        for (int i = 0; i < size; i++) {
            if (element.equals(get(i))) {    //容器中所有的比较操作，都是用equals而不是==
                //将该元素移除
                remove(i);
            }
        }
    }

    public void remove(int index) {
        int numMoved = elementData.length - index - 1;
        if (numMoved > 0) {
            System.arraycopy(elementData, index + 1, elementData, index, elementData.length - index - 1);
        }
        elementData[--size] = null;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("[");
        for (int i = 0; i < size; i++) {
            sb.append(elementData[i] + ",");
        }
        sb.setCharAt(sb.length() - 1, ']');
        return sb.toString();
    }

    public static void main(String[] args) {
        MyArrayList sxtArrayList = new MyArrayList(20);
        for (int i = 0; i < 40; i++) {
            sxtArrayList.add("xiao" + i);
        }
        System.out.println(sxtArrayList);
        System.out.println(sxtArrayList.get(41));
    }
}