手动实现动态数组【练习】

@SuppressWarnings("unchecked")
public class ArrayList<E> {
	private int size;		// 元素的数量	
	private E[] elements; 	// 所有的元素

	private static final int DEFAULT_CAPACITY = 10; // 初始容量
	
	private static final int ELEMENT_NOT_FOUND = -1; // 不存在返回
	
	// 容量小于10一律扩充为10
	public ArrayList(int capacity) { 
		capacity = (capacity < DEFAULT_CAPACITY) ? DEFAULT_CAPACITY : capacity;
		elements = (E[])new Object[capacity];
	}

	// 未传入参数，使用初始容量创建数组
	public ArrayList(){
		this(DEFAULT_CAPACITY);
	}
	
	/**
	 * 元素的数量
	 * @return
	 */
	public int size(){
		return size;
	}
	
	/**
	 * 是否为空
	 * @return
	 */
	public boolean isEmpty(){
		return size == 0;
	}
	
	/**
	 * 是否包含某个元素
	 * @param element
	 * @return
	 */
	public boolean contains(E element){
	
		// 找的到该元素则返回True
		return indexOf(element) != ELEMENT_NOT_FOUND; 
	}
	
	/**
	 * 在index位置插入一个元素
	 * @param index
	 * @param element
	 */
	public void add(int index, E element){ 
		rangeCheckForAdd(index); // 检查下标越界
		ensureCapacity(size + 1); // 确保容量够大
		
		// 先从后往前开始, 将每个元素往后移一位, 然后再赋值
		for (int i = size - 1; i > index; i--) {
			elements[i + 1] = elements[i];
		}
		elements[index] = element;
		size++;
	}
	
	/**
	 * 添加元素到数组最后
	 */
	public void add(E element){
		add(size, element);
	}
	
	/**
	 * 获取index位置的元素
	 * @param index
	 * @param element
	 * @return 原来的元素ֵ
	 */
	public E get(int index){
		rangeCheck(index);
		return elements[index];
	}
	
	/**
	 * 设置index位置的元素
	 * @param index
	 * @param element
	 * @return 原来的元素ֵ
	 */
	public E set(int index, E element){
		// 检查index是否合法
		rangeCheck(index);
	
		E old = elements[index];
		elements[index] = element;
		return old;
	}
	
	/**
	 * 删除index位置的元素
	 * @param index
	 * @return
	 */
	public E remove(int index){
		// 检查index是否合法
		rangeCheck(index);

		// 从前往后开始移, 用后面的元素覆盖前面的元素
		E old = elements[index];
		for (int i = index; i < size - 1; i++) {
			elements[i] = elements[i + 1];
		}
		
		// 删除元素后, 将最后一位设置为null
		elements[--size] = null; 

		// 返回被删除的元素
		return old;
	}
	
	/**
	 * 查看元素的索引
	 * @param element
	 * @return
	 */
	public int indexOf(E element){
		/*
		 不对 null 进行处理也可以，但是健壮性不够
		for (int i = 0; i < size; i++) {
			if(elements[i].equals(element)) return i;
		}
		 */
		 
		 // 对 null 进行处理
		if(element == null){ 
			for (int i = 0; i < size; i++) {
				// 找到后返回下标索引
				if(elements[i] == null) return i;
			}
		}else{
			for (int i = 0; i < size; i++) {
				// 找到后返回下标索引
				if(elements[i].equals(element)) return i;
			}
		}
		
		// 到这里表示找不到
		return ELEMENT_NOT_FOUND;
	}
	
	/**
	 * 清除所有元素
	 */
	public void clear(){
		// 使用泛型数组后要注意内存管理(将元素置null)
		// java官方也依次将每个元素赋值为null
		for (int i = 0; i < size; i++) {
			elements[i] = null;
		}
		size = 0;
	}
	
	/**
	 * 扩容操作
	 */
	private void ensureCapacity(int capacity){
		// 首先获取旧的容量
		int oldCapacity = elements.length;
		
		// 如果旧的容量大于或者等于传入的容量，直接返回
		// 等于的时候刚好还有一个位置
		if(oldCapacity >= capacity) return;
		
		// 新容量为旧容量的1.5倍
		int newCapacity = oldCapacity + (oldCapacity >> 1);
		
		// 用于存储旧容量中的元素值
		E[] newElements = (E[])new Object[newCapacity];
		for (int i = 0; i < size; i++) {
			newElements[i] = elements[i]; // 拷贝原数组元素到新数组
		}
		
		elements = newElements;
		System.out.println("size="+oldCapacity+", 扩容到了"+newCapacity);
	}
	

	// 下标越界抛出的异常，在多处用到，直接提取出来
	private void outOfBounds(int index) {
		throw new IndexOutOfBoundsException("Index:" + index + ", Size:" + size);
	}
	
	// 检查下标越界
	private void rangeCheck(int index){
		if(index < 0 || index >= size){
			outOfBounds(index);
		}
	}
	
	// 检查add()的下标越界(可以在size位置添加元素)
	private void rangeCheckForAdd(int index) {
		if (index < 0 || index > size) {
			outOfBounds(index);
		}
	}
	

	// 打印输出测试
	@Override
	public String toString() {
		
		StringBuilder string = new StringBuilder();
		string.append("size=").append(size).append(", [");
		for (int i = 0; i < size; i++) {
			if(0 != i) string.append(", ");
			string.append(elements[i]);
		}
		string.append("]");
		return string.toString();
	}
}