洛谷题解 P1739 表达式括号匹配

题目描述

假设一个表达式有英文字母（小写）、运算符（+，—，*，/）和左右小（圆）括号构成，以“@”作为表达式的结束符。请编写一个程序检查表达式中的左右圆括号是否匹配，若匹配，则返回“YES”；否则返回“NO”。表达式长度小于255，左圆括号少于20个。
输入格式
一行：表达式
输出格式
一行：“YES” 或“NO”
输入输出样例
输入 #1
2(x+y)/(1-x)@
输出 #1
YES
输入 #2
(25+x)(a*(a+b+b)@
输出 #2
NO
说明/提示
表达式长度小于255，左圆括号少于20个

解题方法：

我们可以使用栈来解题。如果读入一个左括号，则一定有一个右括号与其匹配。那么，如果有左括号，把左括号压入栈。如果有右括号，那么把栈顶的左括号弹出栈。到最后，如果栈中什么都没有，说明表达式括号匹配。否则，如果栈有剩余的括号，就不匹配，例如序列(()。如果栈中已经没有括号了，但是读入一个右括号，也是出错的，例如序列())。

代码

#include<bits/stdc++.h>
using namespace std;
int main(){
    stack<char>stk;
    int c;
    for(;;){
        c=getchar();
        if(c=='@'){
            if(stk.size()==0)cout<<"YES";
            else cout<<"NO";
            return 0;
        }
        else if(c=='(')stk.push(c);
        else if(c==')'){
            if(stk.size()==0){
                cout<<"NO";
                return 0;
            }
            if(stk.top()=='(')stk.pop();
        }
    }
}