数据结构与算法（四）

栈的原理

package com.bjsxt.shujujiegou;
/**栈的原理*/
public class MyStack {

    //栈的底层我们使用数组来看存储数据
    int[] elements;
    public MyStack(){
        elements = new int[0];
    }

    //压入元素


    public void push(int element){
        //创建一个新的数组
        int[] newArr = new int[elements.length + 1];
        //把原数组中的元素复制到新数组中
        for(int i = 0 ; i < elements.length; i++){
            newArr[i] = elements[i];
        }
        //把添加的元素放入新数组中
        newArr[elements.length] = element;
        //使用新数组替换旧数组
        elements = newArr;
    }

    //取出栈顶元素
    public int pop(){
        //栈中没有元素
        if(elements.length==0){
            throw new RuntimeException("stack is empty");
        }
        //取出数组的最后一个元素
        int element = elements[elements.length-1];
        //创建一个新的数组
        int[] newArr = new int[elements.length-1];
        //原数组中除了最后一个元素的其他元素都放在新的数组中
        for(int i=0;i<elements.length-1;i++){
            newArr[i] = elements[i];
        }
        //替换数组
        elements = newArr;
        //返回栈顶元素
        return element;
    }

    //查看栈顶元素
    public int peek(){
        //栈中没有元素
        if(elements.length==0){
            throw new RuntimeException("stack is empty");
        }
        return elements[elements.length-1];
    }

    //判断栈是否为空
    public boolean isEmpty(){
        return elements.length==0;
    }

}

对栈的先进后出进行代码测试

package com.bjsxt.shujujiegou;
//测试TestMyStack
public class TestMyStack {

    public static void main(String[] args){

        //创建一个栈
        MyStack ms = new MyStack();

//        int pop = ms.pop();
//        System.out.println(pop);

        //压入数组                    输入值为：9876543210
        ms.push(9);
        ms.push(8);
        ms.push(7);
        ms.push(6);
        ms.push(5);
        ms.push(4);
        ms.push(3);
        ms.push(2);
        ms.push(1);
        ms.push(0);

        //弹出栈顶元素                 结果为：0123456789
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());
//        System.out.println(ms.pop());   //全部取出报错  数组是空的


        //查看栈顶元素
//        System.out.println(ms.peek());
//        ms.pop();
//        System.out.println(ms.peek());

        //查看栈是否为空

        System.out.println(ms.isEmpty());

    }
}