题解 | #四则运算#

const readline = require('readline')
const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout
})

// 根据栈顶运算符，与栈顶两个元素进行运算
function compute(st1,st2){
  let b = st1.pop()
  let a = st1.pop()
  let op = st2.pop() // 栈顶运算符
  if(op === '+'){
    a = a + b
  }else if(op === '-'){
    a = a - b
  }else if(op === '*'){
    a = a * b
  }else if(op === '/'){
    a = a / b
  }
  st1.push(a)
}

// 比较运算符优先级
function priority(m,n){
  // 如果当前栈顶为 * 或 / ，优先执行该运算符
  if(!m){
    return false
  } else if(m === '('){
      return false
  }else if(m === '+' || m === '-' && n === '*' || n === '/'){
    return false
  }
  return true
}

function test(s){
  let stack1 = [] // 记录运算数字
  let stack2 = [] // 记录运算符
  let flag = false // flag用于标记每轮是否出现过数字， 数字之前的+-为正负号，数字之后的为加减号
  for(let i=0;i<s.length;i++){
    // 碰到 ( [ { 左括号，就将右括号 ) 压入栈中
    if(s[i] === '(' || s[i] === '[' || s[i] === '{'){
      stack2.push('(')
    }else if(s[i] === ')' || s[i] === ']' || s[i] === '}'){ // 遇到右括号
      while(stack2[stack2.length-1] !== '('){ // 弹出开始计算直到遇到左括号
        compute(stack1,stack2)
      }
      stack2.pop() // 弹出左括号
    }else if(flag){ // 运算符
      while(priority(stack2[stack2.length-1],s[i])){ // 比较运算符优先级
        compute(stack1,stack2) // 可以直接计算
      }
      stack2.push(s[i])
      flag = false
    }else{ // 数字
      let j = i
      if(s[i] === '+' || s[i] === '-'){ // 正负号
        i++
      }
      // 如果截取部分为负数或者正数，直接推入栈中，如果符号为 * 或 / ，则i--，将flag标记掷为true，方便下次进入运算符判断中
      while(/\d/.test(s[i])){
        i++
      }
      let temp = s.substr(j,i-j)
      stack1.push(Number(temp))
      i--
      flag = true // 数字结束，下一次flag为true就是运算符了
    }
  }
  while(stack1.length > 1){
    compute(stack1,stack2)
  }
  console.log(stack1.pop())
}

rl.on('line',function(line){
  test(line)
})