1. 程式人生 > >JAVA面試題 手寫ArrayList的實現,在筆試中過關斬將?

JAVA面試題 手寫ArrayList的實現,在筆試中過關斬將?

面試官Q1:可以手寫一個ArrayList的簡單實現嗎?

我們都知道ArrayList是基於陣列實現,如果讓你實現JDK原始碼ArrayList中add()、remove()、get()方法,你知道如何實現嗎?這一節,我們不看原始碼,我們想想如何簡單的實現ArrayList幾個基本方法?

 

確定資料結構

我們知道,ArrayList中無論什麼資料都能放,是不是意味著它是一個Object型別,既然是陣列,那麼是不是Object[]陣列型別的?所以我們定義的資料結構如下:

private Object[] elementData;  
private int size;

設定自定義的MyArrayList的長度為10

 public MyArrayList(){
         this(10);
 }
 
public MyArrayList(int initialCapacity){
      if(initialCapacity<0){
             try {
                 throw new Exception();
             } catch (Exception e) {
                e.printStackTrace();
            }
        }
        elementData = new Object[initialCapacity];
}

有了存放資料的位置,接下來,我們想想如何將資料放入陣列?

 

新增資料

public void add(Object obj){
        elementData[size++]=obj;
}

每新增一個元素,size就會自增1,我們定義的陣列長度為10,當我們新增到11個元素的時候,顯然沒有地方存放新新增的資料,這個時候我們需要對陣列進行擴容處理對上面程式碼做如下修改:

 public void add(Object obj){
         if(size==elementData.length){
            //建立一個新的陣列,並且這個陣列的長度是原陣列長度的2倍
             Object[] newArray = new Object[size*2];
            //使用底層拷貝,將原陣列的內容拷貝到新陣列
             System.arraycopy(elementData, 0, newArray, 0, elementData.length);
            //並將新陣列賦值給原陣列的引用
             elementData = newArray;
         }
        //新來的元素,直接賦值
        elementData[size++]=obj;
}

用一張圖來表示就是這樣的:

  

查詢資料

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

 

刪除資料

接著我們看一下刪除的操作。ArrayList支援兩種刪除方式:

  • 按照下標刪除

  • 按照元素刪除,這會刪除ArrayList中與指定要刪除的元素匹配的第一個元素

對於ArrayList來說,這兩種刪除的方法差不多,都是呼叫的下面一段程式碼:

 public void remove(int index){
         //刪除指定位置的物件
         //a b d e
         int numMoved = size - index - 1;
         if (numMoved > 0){
             System.arraycopy(elementData, index+1, elementData, index,
                     numMoved);
         }
         elementData[--size] = null; // Let gc do its work
}

其實做的事情就是兩件:

  • 把指定元素後面位置的所有元素,利用System.arraycopy方法整體向前移動一個位置

  • 最後一個位置的元素指定為null,這樣讓gc可以去回收它

用圖表示是這樣的:

指定位置新增資料

把從指定位置開始的所有元素利用System,arraycopy方法做一個整體的複製,向後移動一個位置(當然先要用ensureCapacity方法進行判斷,加了一個元素之後陣列會不會不夠大),然後指定位置的元素設定為需要插入的元素,完成了一次插入的操作。

public void add(int index,Object obj){
        ensureCapacity();  //陣列擴容
        System.arraycopy(elementData, index, elementData, index + 1,
                 size - index);
        elementData[index] = obj;
        size++;
}

用圖表示這個過程是這樣的:

  

  

總結一下

從上面的幾個過程總結一下ArrayList的特點:

  • ArrayList底層以陣列實現,是一種隨機訪問模式,通過下標索引定位資料,所以查詢非常快

  • ArrayList在順序新增一個元素的時候非常方便,只是往數組裡面添加了一個元素而已(這裡指的末尾新增資料)

  • 當刪除元素的時候,涉及到一次元素複製移位,如果要複製的元素很多,那麼就會比較耗費效能

  • 當插入元素的時候,涉及到一次元素複製移位,如果要複製的元素很多,那麼就會比較耗費效能

因此,ArrayList比較適合順序新增、隨機訪問的場景。