括弧匹配检验

题目描述

假设表达式中允许包含两种括号：圆括号和方括号，其嵌套的顺序随意，如（［ ］（））或［（［ ］［ ］）］等为正确的匹配，［（ ］）或（［ ］（ ）或 ( ( ) ) ）均为错误的匹配。

现在的问题是，要求检验一个给定表达式中的括弧是否正确匹配？

输入一个只包含圆括号和方括号的字符串，判断字符串中的括号是否匹配，匹配就输出 “OK” ，不匹配就输出“Wrong”。输入一个字符串：［（［］［］）］，输出：OK。

输入

输入仅一行字符（字符个数小于255）。

输出

匹配就输出 “OK” ，不匹配就输出“Wrong”。

样例

输入数据 1

［（］）

Copy

输出数据 1

Wrong

Copy

来源

一本通在线评测

代码

#include<bits/stdc++.h>
using namespace std;
string s;
stack<char>p;
int main(){
    getline(cin,s);
    for(int i=0;s[i];i++){
        if(s[i]=='(')
            p.push(s[i]);
        else if(s[i]=='[')
            p.push(s[i]);
        else if(s[i]==')'||s[i]==']'){
            if(!p.empty()){
                if(s[i]==']'){
                    if(p.top()!='['){
                        cout<<"Wrong"<<endl;
                        return 0;
                    }
                    else p.pop();
                }else{
                    if(s[i]!=')'){
                        cout<<"Wrong"<<endl;
                        return 0;
                    }
                    else p.pop();
                }
            }else{
                cout<<"Wrong"<<endl;
                return 0;
            }
        }
    }
    if(p.empty())cout<<"OK";
    else cout<<"Wrong";
    return 0;
}