数据结构之数组下的栈

数组下的栈

特点：只能从栈顶添加元素，也只能从栈顶取出元素(后进先出)。

自定义栈接口

package com.company.stack;

/**
 * 自定义栈接口
 *  特点：先进后出
 * @param <T>
 */
public interface Stack<T> {

    // 获取栈长度
    int getSize();

    // 进栈操作
    void push(T ele) throws Exception;

    // 出栈操作
    T pop();

    // 查看栈顶元素
    T peek();

    // 判断栈是否为空
    boolean isEmpty();
}

自定义栈的实现类

package com.company.stack;

import com.company.array.Array;

/**
 * 自定义栈结构
 * 特点：只能从栈顶添加元素，也只能从栈顶取出元素(后进先出)
 *
 * @param <T> 泛型
 * @Author: wenhua
 * @CreateTime: 2023-01-07  07:43
 */
public class MyStack<T> implements Stack<T> {

    private Array<T> array;

    public MyStack() {
        this(10);// 调用有参构造函数
    }

    public MyStack(int capacity) {
        array = new Array<>(capacity);
    }

    /**
     * 判断栈是否为空
     *
     * @return
     */
    @Override
    public boolean isEmpty() {
        return array.isEmpty();
    }

    /**
     * 返回栈元素个数
     *
     * @return
     */
    @Override
    public int getSize() {
        return array.getSize();
    }

    /**
     * 进栈操作
     *
     * @param ele
     */
    @Override
    public void push(T ele) {
        try {
            array.addTail(ele);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    /**
     * 出栈操作
     *
     * @return
     */
    @Override
    public T pop() {
        T result = null;
        try {
            result = array.removeTailElement();
        } catch (Exception e) {
            System.out.println("index is error");
        }
        return result;
    }

    /**
     * 查看栈顶元素
     *
     * @return
     */
    @Override
    public T peek() {
        T result = null;
        try {
            result = array.getElementByIndex(array.getSize() - 1);
        } catch (Exception e) {
            System.out.println("index is error");
        }
        return result;
    }

    @Override
    public String toString() {
        StringBuffer sbf = new StringBuffer();
        sbf.append("栈中总共有：" + getSize() + "个元素:");// '\n'表示换行
        sbf.append("[");

        try {
            for (int i = getSize() - 1; i >= 0; i--) {
                sbf.append(array.getElementByIndex(i));
                if (i != 0) {
                    sbf.append(",");
                }
            }
            sbf.append("]");
            return sbf.toString();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}

main方法测试

package com.company.stack;

/**
 * @Author: wenhua
 * @CreateTime: 2023-01-07  07:58
 */
public class Main {
    public static void main(String[] args) {
        MyStack<String> stack = new MyStack<String>(2);
        stack.push("Jack");
        System.out.println("栈中元素个数为:" + stack.getSize());
        System.out.println("判断栈是否为空:" + stack.isEmpty());
        System.out.println("查看栈顶元素:" + stack.peek());
        stack.push("Jim");
        stack.push("wenhua");
        System.out.println(stack);
        while (!stack.isEmpty()) {
            System.out.println("pop出栈操作：" + stack.pop());
            System.out.println(stack);
            System.out.println("-------------");
        }
        System.out.println("判断栈是否为空:" + stack.isEmpty());
        // 栈为空，返回异常处理结果
        // System.out.println(myStack.pop()); 异常反馈 index is error 返回null
    }
}

添加完数据后，数组下的栈结构如下图：

运行结果

栈中元素个数为:1
判断栈是否为空:false
查看栈顶元素:Jack
栈中总共有：3个元素:[wenhua,Jim,Jack]
pop出栈操作：wenhua
栈中总共有：2个元素:[Jim,Jack]
-------------
pop出栈操作：Jim
栈中总共有：1个元素:[Jack]
-------------
pop出栈操作：Jack
栈中总共有：0个元素:[]
-------------
判断栈是否为空:true

注意：

栈在循环弹出时，长度是可变的，所以不能通过简单的for(int i = 0;i<stack.size();i++)来进行判断，只能通过while(!stack.isEmpty())或while(stack.size()!=0)来判断。