X1998-表达式括号匹配—信息学奥数比赛官方发布

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

输入
一行数据，即表达式。

输出
一行，即“YES” 或“NO”。

样例
输入数据 1

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

输出数据 1

YES

代码：

#include<bits/stdc++.h>
using namespace std;
stack<char> stk;
int main()
{
    string str;
    cin>>str;
    for(int i=0;i<str.size();i++)
    {
        if(str[i]=='@') break;
        if(str[i]=='(') stk.push(str[i]);
        if(str[i]==')')
        {
            if(stk.size()==0)
            {
                cout<<"NO";
                return 0;
            }
            else stk.pop();
        }
    }
    if(stk.size()) cout<<"NO";
    else cout<<"YES";
    return 0;
}