Java数据结构与算法之使用栈完成表达式的计算的小练习

使用栈完成表达式的计算思路：
1、通过一个索引值index，来遍历我们的表达式
2、我们一共创建两个栈，一个树栈numStack，用于存放数据，另一个是符号栈operStack，用于存放运算符
3、如果我们发现是一个数字，就直接入树栈
4、如果发现扫描到的是一个符号，就分如下情况：
4.1如果发现当前的符号栈为空，就直接入栈
4.2如果符号栈有操作符，就进行比较，如果当前的操作符的优先级小于或等于栈中的操作符，
就需要从树栈中pop出两个树，再从符号栈中pop出一个符号，进行运算，将得到的结果，入树栈，
然后将当前的操作符入符号栈，如果当前的操作符的优先级大于栈中的操作符，则直接入符号栈
5、当表达式扫描完毕，就顺序的从树栈和符号栈中pop出相应的数和符号
6、最后在树栈中只有一个数字，就是表达式的结果

package com.wrx.dataStructures.stack;
/**
 * @auther wrx
 * @time 2022-09-21-14:40
 */
/*
使用栈完成表达式的计算思路：
1、通过一个索引值index，来遍历我们的表达式
2、我们一共创建两个栈，一个树栈numStack，用于存放数据，另一个是符号栈operStack，用于存放运算符
3、如果我们发现是一个数字，就直接入树栈
4、如果发现扫描到的是一个符号，就分如下情况：
4.1如果发现当前的符号栈为空，就直接入栈
4.2如果符号栈有操作符，就进行比较，如果当前的操作符的优先级小于或等于栈中的操作符，
就需要从树栈中pop出两个树，再从符号栈中pop出一个符号，进行运算，将得到的结果，入树栈，
然后将当前的操作符入符号栈，如果当前的操作符的优先级大于栈中的操作符，则直接入符号栈
5、当表达式扫描完毕，就顺序的从树栈和符号栈中pop出相应的数和符号
6、最后在树栈中只有一个数字，就是表达式的结果
 */
public class Calculator {
    public static void main(String[] args) {
        String expression="3+2*6-2";
        //创建两个栈，一个树栈，一个符号栈
        ArrayStack2 numStack=new ArrayStack2(10);
        ArrayStack2 operStack=new ArrayStack2(10);
        //定义需要的相关变量
        int index=0;
        int num1=0;
        int num2=0;
        int oper=0;
        int res=0;
        char ch=' ';
        //开始用while循环扫描expresion
        while(true){
            //依次得到expression的每一个字符
            ch=expression.substring(index,index+1).charAt(0);
            //判断ch是数字还是运算符号，然后做相应的处理
            if(operStack.isOper(ch)){//如果是一个运算符
                //判断当前的符号栈是否为空
                if(!operStack.isEmpty()){
                    //如果符号栈有操作符，就进行比较，如果当前的操作符的优先级小于或等于栈中的操作符，
                    //就需要从树栈中pop出两个树，再从符号栈中pop出一个符号，进行运算，将得到的结果，入树栈，
                    if(operStack.priority(ch)<=operStack.priority(operStack.peek())){//当前的操作符的优先级小于或等于栈中的操作符
                        num1=numStack.pop();
                        num2=numStack.pop();
                        oper=operStack.pop();
                        res=numStack.cal(num1,num2,oper);
                        //把运算结果入树栈
                        numStack.push(res);
                        //然后将当前的操作符入符号栈
                        operStack.push(ch);
                    }else {//如果当前的操作符的优先级大于栈中的操作符，则直接入符号栈
                        operStack.push(ch);
                    }
                }else {//当前符号栈为空，直接放入运算符
                    operStack.push(ch);
                }
            }
            else {//不是运算符，是数字，则直接入树栈
                numStack.push(ch-48);
            }
            //让index+1，并判断是否扫描到expression最后
            index++;
            if(index>=expression.length()){//此时已经到达运算式的最后
                break;
            }
        }
        //当表达式扫描完毕，就顺序的从树栈和符号栈中pop出相应的数和符号
        while (true){
            //如果符号栈为空，则计算到最后的结果，树栈中只有一个数字
            if(operStack.isEmpty()){
                break;
            }
            num1=numStack.pop();
            num2=numStack.pop();
            oper=operStack.pop();
            res=numStack.cal(num1,num2,oper);
            numStack.push(res);
        }
        System.out.println("表达式"+expression+"的值为"+numStack.pop());
    }
}
//先创建一个栈，直接使用全面创建好的
//定义一个ArrayStack2表示栈，需要扩展一下功能
class ArrayStack2{
    private int maxSize;//栈的大小
    private int[] stack;//数组模拟栈，数据就放在该数组中
    private int top=-1;//top表示栈顶
    //构造器，构造栈的大小
    public ArrayStack2(int maxSize) {
        this.maxSize = maxSize;
        stack = new int[this.maxSize];
    }
    //栈满
    public boolean isFull(){
        return top==maxSize-1;
    }
    //栈空
    public boolean isEmpty(){
        return top==-1;
    }
    //入栈——push
    public void push(int value){
        //先判断栈是否满
        if(isFull()){
            System.out.println("栈已满");
            return;
        }
        top++;
        stack[top]=value;
    }
    //出栈——pop，将栈顶的数据返回
    public int pop(){
        //先判断栈是否为空
        if (isEmpty()){
            throw new RuntimeException("栈空，没有数据");
        }
        int value=stack[top];
        top--;
        return value;
    }
    //显示栈的情况，即遍历栈，遍历时需要从栈顶开始显示数据
    public void list(){
        if(isEmpty()){
            System.out.println("栈空，没有数据");
            return;
        }
        //需要从栈顶开始显示数据
        for(int i=top;i>=0;i--){
            System.out.println("stack["+i+"]"+stack[i]);
        }
    }
    //返回运算符的优先级，优先级是由程序员来确定，优先级使用数字来表示，数字越大，表示优先级越高
    public int priority(int oper){
        if(oper=='*'||oper=='/'){
            return 1;
        } else if (oper=='+'||oper=='-') {
            return 0;
        }else {
            return -1;
        }
    }
    //判断是不是一个运算符
    public boolean isOper(char val){
        return val=='+'||val=='-'||val=='*'||val=='/';
    }
    //返回当前栈顶的值，但不是真正的pop
    public int peek(){
        return stack[top];
    }
    //计算方法
    public int cal(int num1,int num2,int oper){
        int res=0;//res用于存放计算的结果
        switch (oper) {
            case '+':
                res=num1+num2;
                break;
            case '-':
                res=num2-num1;
                break;
            case  '*':
                res=num1*num2;
                break;
            case '/':
                res=num2/num1;
                break;
            default:
                break;
        }
        return res;
    }
}