括号匹配（带优先级）栈C++

菜鸡小陈2024-01-16 14:54

在算术表达式中，除了加、减、乘、除等运算外，往往还有括号。包括有大括号 {}，中括号 \[\]，小括号 ()，尖括号 <> 等。
对于每一对括号，必须先左边括号，然后右边括号；如果有多个括号，则每种类型的左括号和右括号的个数必须相等；对于多重括号的情形，按运算规则，从外到内的括号嵌套顺序为：大括号->中括号->小括号->尖括号，另外相同的括号可以嵌套。
例如，{ $()$ },{(())},{{}} 为一个合法的表达式，而 ( ${}$ ),{(\[\])}, ${\<\>}$ 都是非法的。

输入格式

第一行包含整数 n，表示共有 n 个表达式需要判断。接下来 n行，每行包含一个括号表达式。

输出格式

每行输出一个表达式的判断结果。如果合法输出 YES，否则输出 NO。

数据范围

1≤n≤100

表达式长度不超过 100。

输入样例：

{ $(\<\>)$ }

()

<>()\[\]{}

{}

{()}

输出样例：

YES

cpp 复制代码

#include<iostream>
#include<stack>
#include<unordered_map>
using namespace std;
unordered_map<char,int> mp={
    {'{',4},
    {'}',-4},
    {'[',3},
    {']',-3},
    {'(',2},
    {')',-2},
    {'<',1},
    {'>',-1}
};
string str;
bool check()
{
    stack<int> stk;
    for(int i=0;i<str.size();i++)
    {
        int t=mp[str[i]];
        if(t<0)
        {
            if(stk.empty()||stk.top()!=-t) return false;
            else stk.pop();
        }
        else
        {
            if(stk.size()&&t>stk.top()) return false;
            stk.push(t);
        }
    }
    return stk.empty();
}
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        cin>>str;
        if(check()) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}