P1713 AC 代码

题目描述

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

输入格式

一行：表达式。

输出格式

一行：YES 或 NO。

输入输出样例

输入 #1复制

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

输出 #1复制

YES

输入 #2复制

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

输出 #2复制

NO

说明/提示

表达式长度小于 255255，左圆括号少于 2020 个。

#include <iostream>
#include <stack>
using namespace std;

int main() {
    string s;
    cin >> s;
    stack<char> stk;
    bool flag = true;
    for (int i = 0; i < s.length() && flag; i++) {
        if (s[i] == '(') {
            stk.push(s[i]);
        } else if (s[i] == ')') {
            if (stk.empty()) {
                flag = false;
            } else {
                stk.pop();
            }
        }
    }
    if (!flag || !stk.empty()) {
        cout << "NO" << endl;
    } else {
        cout << "YES" << endl;
    }
    return 0;
}