在使用java语言编写程序时,ArrayList是我们经常使用的工具,因为它功能强大,可以储存任何引用类型数据,同时又可以灵活变化长度。下面我们来讨论如何自己编写一个简易的ArrayList。
public class MyArrayList<E> {
private E[] data; //定义数组
private int size;//长度
@SuppressWarnings("unchecked")
public MyArrayList(int capacity) { //有参构造设置初始长度
data=(E[]) new Object[capacity];
size=0;
}
@SuppressWarnings("unchecked")
public MyArrayList(){ //无参构造,默认初始长度为0
data=(E[]) new Object[0];
size=0;
}
public void add(int index,E obj) {
//长度不够则扩容
if(size==data.length){
if(size<=1){
resize(data.length+1);
}else{
resize((int)(1.5*data.length)); //扩容1.5倍
}
}
if(index<0||index>size) {
throw new IllegalArgumentException("add failed");//抛出非法异常
}
for(int i=size-1;i>=index;i--)
data[i+1]=data[i];
data[index]=obj;
size++;
}
public void add(E obj){
if(size==data.length){
if(size<=1){
resize(data.length+1);
}else{
resize((int)(1.5*data.length));
}
}
data[size]=obj;
size++;
}
void set(int index,E e) {
if(index<0||index>=size)
throw new IllegalArgumentException("add failed");
data[index]=e;
}
public E get(int index) {
if(index<0||index>=size)
throw new IllegalArgumentException("get failed");
return data[index];
}
public E remove(int index) {
if(index<0||index>=size) {
throw new IllegalArgumentException("Error");
}
E ret=data[index];
for(int i=index+1;i<size;i++) //遍历移动数组元素实现删除
data[i-1]=data[i];
size--;
if(size==data.length/4&&data.length/2!=0)
resize(data.length/2);
return ret;
}
public void remove(E obj) {
int index=-1;
for(int i=0;i<size;i++) {
if(data[i].equals(obj)){
index=i;
}
}
if(index!=-1){
remove(index);
}
}
public void resize(int newCapacity) {
@SuppressWarnings("unchecked")
E[] newdata=(E[]) new Object[newCapacity];
for(int i=0;i<data.length;i++) {
newdata[i]=data[i]; //将数组内容拷贝到新数组
}
data=newdata;
}
public int size(){
return size;
}
}