括弧匹配检验

题目来源：http://ybt.ssoier.cn:8088/problem_show.php?pid=1354

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 11190 通过数: 3542
【题目描述】
假设表达式中允许包含两种括号：圆括号和方括号，其嵌套的顺序随意，如（［ ］（））或［（［ ］［ ］）］等为正确的匹配，［（ ］）或（［ ］（ ）或 ( ( ) ) ）均为错误的匹配。

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

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

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

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

【输入样例】
［（］）
【输出样例】
Wrong
【来源】

No

知识点：栈的应用，括号匹配。

#include<cstdio>
#include<iostream>
#include<stack>
using namespace std;

stack<char> s;

int main(){
    string str;
    cin>>str;
    int len=str.length();
    while(!s.empty()){
        s.pop();
    }
    for(int i=0;i<len;i++){
        if(str[i]=='<' || str[i]=='(' || str[i]=='[' || str[i]=='{'){
            s.push(str[i]);
        }
        else{
            if(s.empty()){
                cout<<"Wrong"<<endl;
                return 0;
            }
            char tp=s.top();
            switch(str[i]){
                case '>':
                if(tp!='<'){
                    cout<<"Wrong"<<endl;
                    return 0;
                }
                break;
                case ')':
                if(tp!='('){
                    cout<<"Wrong"<<endl;
                    return 0;
                }
                break;
                case ']':
                if(tp!='['){
                    cout<<"Wrong"<<endl;
                    return 0;
                }
                break;
                default:
                if(tp!='{'){
                    cout<<"Wrong"<<endl;
                    return 0;
                }
                break;
            }
            s.pop();
        }
    }
    if(!s.empty()){
        cout<<"Wrong"<<endl;
        return 0;
    }
    else{
        cout<<"OK"<<endl;
        return 0;
    }
}