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]