数组在内存中,是一串连续的内存,针对查找某个下标的元素的情况,时间复杂度为O(1)
但是数组在创建时,就要确定需要申请多少内存,保存的元素数量有限。
动态数组的原理就是当数组的元素达到容量上限是,重新申请一片更大的内存,将原来的数组元素迁移到新的地址空间上。
动态数组通过扩容的形式,使数组的容量不受限制。
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;
public ArrayList(int capacity){
capacity = (capacity < DEFAULT_CAPACITY) ? DEFAULT_CAPACITY : capacity;
elements = (E[]) new Object[capacity];
}
public ArrayList() {
this(DEFAULT_CAPACITY);
}
/**
* 在数组最后添加一个元素
* @param element
*/
public void add(E element) {
add(size, element);
}
/**
* 在指定的位置添加一个元素
* @param index
* @param element
*/
public void add(int index, E element) {
rangeCheckForAdd(index);
ensureCapacity(size + 1);
for (int i = size; i > index; i--) {
elements[i] = elements[i - 1];
}
elements[index] = element;
size++;
}
/**
* 删除某个位置的元素
* @param index
* @return
*/
public E remove(int index) {
rangeCheck(index);
E old = elements[index];
for (int i = index + 1; i < size; i++) {
elements[i - 1] = elements[i];
}
elements[--size] = null;
trim();
return old;
}
/**
* 获取当前元素数量
* @return
*/
public int size() {
return size;
}
/**
* 清空当前数组内元素
*/
public void clear() {
// 将所有元素的内存指向地址设置为空,释放内存
for(int i = 0; i < size; i++) {
elements[i] = null;
}
size = 0;
}
/**
* 获取指定下标的元素
* @param index
* @return
*/
public E get(int index) {
rangeCheck(index);
return elements[index];
}
/**
* 修改指定位置的元素
* @param index
* @param element
* @return
*/
public E set(int index, E element) {
rangeCheck(index);
E old = elements[index];
elements[index] = element;
return old;
}
/**
* 判断当前数组是否为空
* @return
*/
public boolean isEmpty() {
return size == 0;
}
/**
* 查看是否包含某元素
* @param element
* @return
*/
public boolean contains(E element) {
return indexOf(element) != ELEMENT_NOT_FOUND;
}
/**
* 获取某元素在数组中的下标
* @param element
* @return
*/
public int indexOf(E element) {
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 (element.equals(elements[i])) return i;
}
}
return ELEMENT_NOT_FOUND;
}
/**
* 下标越界
* @param index
*/
private void outOfBounds(int index) {
throw new IndexOutOfBoundsException("Index:" + index + ", Size:" + size);
}
/**
* 判断是否需要扩容,保证有足够的容量
* @param capacity
*/
private void ensureCapacity(int capacity) {
int oldCapacity = elements.length;
if (oldCapacity >= capacity) return;
//扩容,增大到原来的1.5倍
int newCapacity = oldCapacity + (oldCapacity >> 1);
E[] newElement = (E[]) new Object[newCapacity];
for (int i = 0; i < size; i++) {
newElement[i] = elements[i];
}
elements = newElement;
System.out.println(oldCapacity + "扩容为 " + newCapacity);
}
private void trim() {
int oldCapacity = elements.length;
int newCapacity = oldCapacity >> 1;
// 当数组内元素小于容量的一半,并且不小于默认容量,将数组的容量缩小为原来的一半
if (size > newCapacity || oldCapacity <= DEFAULT_CAPACITY) return;
E[] newElements = (E[]) new Object[newCapacity];
for (int i = 0; i < size; i++) {
newElements[i] = elements[i];
}
elements = newElements;
}
/**
* 判断是否下标越界
* @param index
*/
private void rangeCheck(int index) {
if(index < 0 || index >= size) {
outOfBounds(index);
}
}
/**
* 判断是否下标越界(用于添加)
* @param index
*/
private void rangeCheckForAdd(int index) {
if(index < 0 || index > size) {
outOfBounds(index);
}
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("size=").append(size).append(", [");
for (int i = 0; i < size; i++) {
if (i != 0) {
sb.append(", ");
}
sb.append(elements[i]);
}
sb.append("]");
return sb.toString();
}
public static void main(String[] args) {
ArrayList<Integer> list = new ArrayList<>();
list.add(11);
list.add(22);
list.add(33);
list.add(44); // 11 22 33 44
list.add(1, 55); // 11 55 22 33 44
list.remove(0); // 55 22 33 44
list.set(2, 66); // 55 22 66 44
// list.clear();
System.out.println(list.indexOf(22));
System.out.println(list);
}
}