数据结构源码
Array数据结构:
public class Array<E> {
private E[] data; // 数组实体,此时data.length为数组总容量
private int size; // 元素个数
/**
* 无参构造函数,默认数组容量size为10
*/
public Array() {
this(10);
}
/**
* 构造函数,传入数组的容量size构造Array
* @param capacity
*/
public Array(int capacity) {
data = (E[]) new Object[capacity];
this.size = 0;
}
public Array(E[] arr) {
data = (E[]) new Object[arr.length];
for (int i = 0; i < arr.length; i++) {
data[i] = arr[i];
}
size = arr.length;
}
/**
* 获取数组元素个数
* @return
*/
public int getSize() {
return size;
}
/**
* 获取数组容量
* @return
*/
public int getCapacity() {
return data.length;
}
/**
* 返回数组是否为空
* @return
*/
public boolean isEmpty() {
return size == 0;
}
/**
* 在插入位置index处插入一个元素e
* @param index
* @param e
*/
public void add(int index, E e) {
if (index < 0 || index > size) {
throw new IllegalArgumentException("add failed. Require index >= 0 and index < size");
}
if (size == data.length)
resize(2 * data.length);
// 所有元素右移,给插入的index处腾出位置
for (int i = size - 1; i >= index; i--) {
data[i + 1] = data[i];
}
// 插入
data[index] = e;
size++;
}
/**
* 向所有元素后添加一个新元素
* @param e
*/
public void addLast(E e) {
add(size, e);
}
/**
* 向所有元素前添加一个新元素
* @param e
*/
public void addFirst(E e) {
add(0, e);
}
/**
* 获取index索引位置的元素
* @param index
* @return
*/
public E get(int index) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal");
}
return data[index];
}
/**
* 修改index索引位置的元素为e
* @param index
* @param e
*/
public void set(int index, E e) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal");
}
data[index] = e;
}
/**
* 查找数组是否有元素e
* @param e
* @return
*/
public boolean contains(E e) {
for (int i = 0; i < size; i++) {
if (data[i].equals(e)) {
return true;
}
}
return false;
}
/**
* 查找数组中元素e所在的索引,如果不存在元素e,返回-1
* @param e
* @return
*/
public int find(E e) {
for (int i = 0; i < size; i++) {
if (data[i].equals(e)) {
return i;
}
}
return -1;
}
/**
* 从数组中删除index位置的元素,返回删除的元素
* @param index
* @return
*/
public E remove(int index) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal.");
}
E ret = data[index];
for (int i = index + 1; i < size; i++) {
data[i - 1] = data[i];
}
size--;
data[size] = null; // loitering objects != memory leak
if (size == data.length / 4 && data.length / 2 != 0) {
resize(data.length / 2);
}
return ret;
}
/**
* 从数组中删除第一个元素,返回删除元素
* @return
*/
public E removeFirst() {
return remove(0);
}
/**
* 从数组中删除最后一个元素,返回删除的元素
* @return
*/
public E removeLast() {
return remove(size - 1);
}
/**
* 从数组中删除元素e
* @param e
*/
public void removeElement(E e) {
int index = find(e); // 找元素e
if (index != -1) {
remove(index);
}
}
@Override
public String toString() {
StringBuilder res = new StringBuilder();
res.append(String.format("Array: size = %d, capacity = %d\n", size, data.length));
res.append('[');
for (int i = 0; i < size; i++) {
res.append(data[i]);
if (i != size - 1) {
res.append(", ");
}
}
res.append(']');
return res.toString();
}
/**
* 将数组空间的容量变成newCapacity大小
* @param newCapacity
*/
private void resize(int newCapacity) {
E[] newData = (E[]) new Object[newCapacity]; // 新数组
for (int i = 0; i < size; i++) {
newData[i] = data[i];
}
data = newData; // 新数组替换老数组
}
}
数据结构拆解
维护字段
private E[] data; // 数组实体,此时data.length为数组总容量
private int size; // 元素个数
构造函数
/**
* 无参构造函数,默认数组容量size为10
*/
public Array() {
this(10);
}
/**
* 构造函数,传入数组的容量size构造Array
* @param capacity
*/
public Array(int capacity) {
data = (E[]) new Object[capacity];
this.size = 0;
}
public Array(E[] arr) {
data = (E[]) new Object[arr.length];
for (int i = 0; i < arr.length; i++) {
data[i] = arr[i];
}
size = arr.length;
}
增
/**
* 在插入位置index处插入一个元素e
* @param index
* @param e
*/
public void add(int index, E e) {
if (index < 0 || index > size) {
throw new IllegalArgumentException("add failed. Require index >= 0 and index < size");
}
if (size == data.length)
resize(2 * data.length);
// 所有元素右移,给插入的index处腾出位置
for (int i = size - 1; i >= index; i--) {
data[i + 1] = data[i];
}
// 插入
data[index] = e;
size++;
}
/**
* 向所有元素后添加一个新元素
* @param e
*/
public void addLast(E e) {
add(size, e);
}
/**
* 向所有元素前添加一个新元素
* @param e
*/
public void addFirst(E e) {
add(0, e);
}
删
/**
* 从数组中删除index位置的元素,返回删除的元素
* @param index
* @return
*/
public E remove(int index) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal.");
}
E ret = data[index];
for (int i = index + 1; i < size; i++) {
data[i - 1] = data[i];
}
size--;
data[size] = null; // loitering objects != memory leak
if (size == data.length / 4 && data.length / 2 != 0) {
resize(data.length / 2);
}
return ret;
}
/**
* 从数组中删除第一个元素,返回删除元素
* @return
*/
public E removeFirst() {
return remove(0);
}
/**
* 从数组中删除最后一个元素,返回删除的元素
* @return
*/
public E removeLast() {
return remove(size - 1);
}
/**
* 从数组中删除元素e
* @param e
*/
public void removeElement(E e) {
int index = find(e); // 找元素e
if (index != -1) {
remove(index);
}
}
改
/**
* 修改index索引位置的元素为e
* @param index
* @param e
*/
public void set(int index, E e) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal");
}
data[index] = e;
}
/**
* 将数组空间的容量变成newCapacity大小
* @param newCapacity
*/
private void resize(int newCapacity) {
E[] newData = (E[]) new Object[newCapacity]; // 新数组
for (int i = 0; i < size; i++) {
newData[i] = data[i];
}
data = newData; // 新数组替换老数组
}
查
/**
* 获取数组元素个数
* @return
*/
public int getSize() {
return size;
}
/**
* 获取数组容量
* @return
*/
public int getCapacity() {
return data.length;
}
/**
* 返回数组是否为空
* @return
*/
public boolean isEmpty() {
return size == 0;
}
/**
* 获取index索引位置的元素
* @param index
* @return
*/
public E get(int index) {
if (index < 0 || index >= size) {
throw new IllegalArgumentException("Get failed. Index is Illeagal");
}
return data[index];
}
public E getLast() {
return get(size - 1);
}
public E getFirst() {
return get(0);
}
/**
* 查找数组是否有元素e
* @param e
* @return
*/
public boolean contains(E e) {
for (int i = 0; i < size; i++) {
if (data[i].equals(e)) {
return true;
}
}
return false;
}
/**
* 查找数组中元素e所在的索引,如果不存在元素e,返回-1
* @param e
* @return
*/
public int find(E e) {
for (int i = 0; i < size; i++) {
if (data[i].equals(e)) {
return i;
}
}
return -1;
}
测试用例
测试用例:
public class Main {
public static void main(String[] args) {
Array<Integer> arr = new Array<>();
for (int i = 0; i < 10; i++) {
arr.addLast(i);
}
System.out.println(arr);
arr.add(1, 100);
System.out.println(arr);
arr.addFirst(-1);
System.out.println(arr);
//[-1, 0, 100, 1, 2, 3, 4, 5, 6, 7, 8, 9
arr.remove(2);
System.out.println(arr);
arr.removeElement(4);
System.out.println(arr);
//[-1, 0, 1, 2, 3, 5, 6, 7, 8, 9
arr.removeFirst();
System.out.println(arr);
}
}