java 用數組簡單實現棧

java 代碼

package org.feng.datastruct;

import java.util.Arrays;

/**
 * java 使用數組簡單實現棧
 * @param <E>
 */
public class MyStack<E> {
    // 底層用數組存儲數據
    private Object[] stack;

    // 默認棧的大小
    private final int DEFAULT_LENGTH = 10;
    // 數組中存儲元素的個數
    private int size;

    public MyStack(int length){
        stack = new Object[length];
    }

    public MyStack(){
        this(10);
    }

    public boolean isEmpty(){
        return size == 0;
    }

    /**
     * 取出頂部元素
     * @return E
     */
    @SuppressWarnings("unchecked")
    public E peek(){
        return isEmpty() ? null : (E)stack[size - 1];
    }

    /**
     * 取出頂部元素，並從棧中刪除
     * @return E
     */
    public E pop(){
        stack[size-- - 1] = null;
        return peek();
    }

    /**
     * 元素壓棧（擴容判斷）
     * @param e 要存儲的元素
     * @return 目標元素
     */
    public E push(E e){
        ensureCapacity(size + 1);
        stack[size ++] = e;
        return e;
    }

    /**
     * 擴容
     * @param size
     */
    private void ensureCapacity(int size) {
        int len = stack.length;
        // 數組長度不夠時
        if(size > len){
            int newLen = 10;
            stack = Arrays.copyOf(stack, newLen);
        }
    }

    @Override
    public String toString() {
        return "MyStack: " + Arrays.toString(stack);
    }

    /**
     * 測試
     * @param args
     */
    public static void main(String[] args) {
        MyStack<Integer> myStack = new MyStack<>();

        myStack.push(1);
        myStack.push(2);
        myStack.push(3);

        System.out.println(myStack);

        System.out.println(myStack.peek());
        System.out.println(myStack.pop());
        System.out.println(myStack.toString());
    }
}

運行結果

MyStack: [1, 2, 3, null, null, null, null, null, null, null]
3
2
MyStack: [1, 2, null, null, null, null, null, null, null, null]