1354：括弧匹配检验

【题目描述】

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

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

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

【输入】

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

【输出】

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

【输入样例】

[(])

【输出样例】

Wrong

#include <iostream>
using namespace std;

const int N = 255;
char a[N] = {0};

int main()
{
    char ch = 0;
    int yuan = 0, zhong = 0;
    int top = 0, cnt = 0;

    while (ch != '@' && cnt < N) {
        ch = cin.get();
        switch (ch) {
            case '(':
                a[top++] = ch;
                yuan++;
                break;
            case ')':
                yuan--;
                if (a[--top] != '(') {
                    ch = '@';
                }
                break;
            case '[':
                a[top++] = ch;
                zhong++;
                break;
            case ']':
                zhong--;
                if (a[--top] != '[') {
                    ch = '@';
                }
                break;
            case '@':
            default:
                ch = '@';
                break;
        }
        cnt++;
    }
    if (zhong == 0 && yuan == 0) {
        cout << "OK";
    } else {
        cout << "Wrong";
    }
    return 0;
}