PTA 括号匹配 (18 分)

• 检查一段C语言代码的小括号( )、 中括号 [ ] 和大括号{ } 是否匹配。

输入格式:

在一行中输入一段C语言代码，长度不超过1000个字符（行末以换行符结束）。

输出格式:

第一行输出左括号的数量和右括号的数量，中间以一个空格间隔。
若括号是匹配的，在第二行打印YES，否则打印NO。

输入样例1:

for(int i=0; i<v; i++){ visited[i] = 0; for(int j=0; j<v; j++) scanf("%d",&(g->Adj[i][j])); }

输出样例1:

8 8
YES

输入样例2:

for(int i=0; i<v; i++) a(i]=0;

输出样例2:

2 2
NO

AC：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
int main()
{
    string s;
    getline(cin, s);
    stack<char> left, right;
    int l = 0, r = 0;
    int flag = 0;
    for (int i = 0; i < s.length();i++)
    {
        if(s[i]=='(' || s[i]=='[' || s[i]=='{'){
            l++;
            left.push(s[i]);
        }
        if(s[i]==')' || s[i]==']' || s[i]=='}'){
            r++;
            right.push(s[i]);
        }
        while(!left.empty() && !right.empty())
        {
            if((left.top()=='(' && right.top()==')') || (left.top()=='[' && right.top()==']') || (left.top()=='{' && right.top()=='}')){
                left.pop();
                right.pop();
            }
            else {
                flag = 1;
                break;
            }
        }
    }
    cout << l << " " << r << endl;
    if(left.empty() && right.empty() && flag!=1)
        cout << "YES" << endl;
        else
            cout << "NO" << endl;
    return 0;
}

---