P1739 表达式括号匹配

题目描述

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

输入格式

一行：表达式

输出格式

一行：“YES” 或“NO”

输入输出样例

 

 

 

 

说明/提示

表达式长度小于255，左圆括号少于20个

#include <bits/stdc++.h>
using namespace std;

stack<char>sta;
int main() {
    ios::sync_with_stdio(0);
    string s;
    cin>>s;
    for (int i=0; i<s.size()&&s[i]!='@'; i++) {
        if(s[i]=='(')sta.push('(');
        else if((s[i]==')'&&sta.empty())||(s[i]==')'&&sta.top()!='(')){cout<<"NO";return 0;}
        else if(s[i]==')'&&sta.top()=='(')sta.pop();
    }
    if(sta.empty())cout<<"YES";
    else cout<<"NO";
    return 0;
}