洛谷 P1739 表达式括号匹配（Java）

题目描述

假设一个表达式有英文字母（小写）、运算符（+，—，*，/）和左右小（圆）括号构成，以“@”作为表达式的结束符。请编写一个程序检查表达式中的左右圆括号是否匹配，若匹配，则返回“YES”；否则返回“NO”。表达式长度小于255，左圆括号少于20个。

输入格式

一行：表达式

输出格式

一行：“YES” 或“NO”

输入输出样例

示例1：

输入：

2*(x+y)/(1-x)@

输出：

YES

示例2：

输入：

(25+x)(a(a+b+b)@

输出：

NO

思路

直接用一个栈考虑

1. 如果是左括号直接入栈
2. 如果是右括号就从栈中弹出一个左括号，如果栈为空就输出 NO
3. 考虑到如果全是左括号或者左括号多，就在最后判断栈是否为空即可

代码

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");
        }
    }
}