【stack】括弧匹配检验

问题 G:
时间限制: 1 Sec 内存限制: 128 MB
[提交] [状态]
题目描述
假设表达式中允许包含两种括号:圆括号和方括号，其嵌套的顺序随意，如( ) 或[([ ][ ])]等为正确的匹配，[( ])或( 或 ( ( ) ) )均为错误的匹配。

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

输入一个只包含圆括号和方括号的字符串，判断字符串中的括号是否匹配，匹配就输出 “OK” ，不匹配就输出“Wrong”。输入一个字符串:[([][])]，输出:OK
输入
仅一行字符，字符个数小于 255。
输出
匹配就输出OK ，不匹配就输出Wrong。
样例输入 Copy
[(])
样例输出 Copy
Wrong

#include<bits/stdc++.h>
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define rep1(i,n) for(int i=1;i<=(int)n;i++)
#define redp(i,n) for(int i=(int)n-1;i>=0;i--)
#define redp1(i,n) for(int i=(int)n;i>=1;i--)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
const int dx[] = { 0, 1, -1, 0, 1, -1, 1, -1 };
const int dy[] = { 1, 0, 0, -1, 1, -1, -1, 1 };
const int MOD = 1e9 + 7;
const int INF=0x3f3f3f3f;
//const double Pi=acos(-1);
bool flag=false;
//----------------------------------------------
const int N=2e6+5;
ll a[200005];
//----------------------------------------------
string s;
stack<char>t;
bool fun(char lf,char rg)
{
    if(lf=='(')
    {
        return(rg==')');
    }
    else if(lf=='[')
    {
        return(rg==']');
    }
}

//----------------------------------------------
int main()
{
    ll  n,i,m,j,k,l,cnt=1;
    ll ans=0,m1=0,m2=0;
    int x,y;
    cin>>s;
    char c;
    rep(i,s.size())
    {
        c=s[i];
        if(c == '(' || c == '[' )
        {
            t.push(c);
        }
        else
        {
            if(t.empty()||!fun(t.top(),c))
            {
                printf("Wrong");
                return 0;
            }
            else if(fun(t.top(),c))
            {
                t.pop();
            }


        }
    }
    if(!t.empty())printf("Wrong");
    else
    {
        printf("OK");
    }
    return 0;

}