洛谷 P1739 表达式括号匹配(Java)
题目描述
假设一个表达式有英文字母(小写)、运算符(+,—,*,/)和左右小(圆)括号构成,以“@”作为表达式的结束符。请编写一个程序检查表达式中的左右圆括号是否匹配,若匹配,则返回“YES”;否则返回“NO”。表达式长度小于255,左圆括号少于20个。
输入格式
一行:表达式
输出格式
一行:“YES” 或“NO”
输入输出样例
示例1:
输入:
2*(x+y)/(1-x)@
输出:
YES
示例2:
输入:
(25+x)(a(a+b+b)@
输出:
NO
思路
直接用一个栈考虑
- 如果是左括号直接入栈
- 如果是右括号就从栈中弹出一个左括号,如果栈为空就输出 NO
- 考虑到如果全是左括号或者左括号多,就在最后判断栈是否为空即可
代码
import java.util.*; public class Main { public static void main(String[] args) { Scanner sc = new Scanner(System.in); Stack<Character> stack = new Stack<>(); String s = sc.next(); boolean flag = true; for (int i = 0; i < s.length(); i++) { char ch = s.charAt(i); if (ch == () { stack.push(ch); } else if (ch == )) { if (stack.isEmpty()) { flag = false; break; } else { stack.pop(); } } } if (flag && stack.isEmpty()) { System.out.println("YES"); } else { System.out.println("NO"); } } }