SWUST#962: 括号匹配问题（栈）

962: 括号匹配问题

题目描述

假设表达式中允许包含两种括号:圆括号和方括号。编写一个算法判断表达式中的括号是否正确配对。

输入

由括号构成的字符串，包含”（“、”）“、”[“和”]“。

输出

如果匹配输出YES，否则输出NO。

样例输入

[([])]

样例输出

YES

#include<iostream>
#include<malloc.h>
#include<string.h>
# define size 50
using namespace std;
typedef struct sta
{
	char data[size];
	int top;
}Stack;
void creat(Stack *&p)
{
	p=(Stack *)malloc(sizeof(Stack));
	p->top=-1;
}
int match(Stack *&p,char str[])
{
	for(int i=0;i<strlen(str);i++)
	{
		if(str[i]=='('||str[i]=='[')
		{
			
			p->top++;
			p->data[p->top]=str[i]; //左括号入栈
		}
		else 
		{
			if(str[i]==')'&&p->data[p->top]=='(')
			{
				p->top--; //左右括号匹配出栈
			}
			else if(str[i]==']'&&p->data[p->top]=='[')
			{
				p->top--; //左右括号匹配出栈
			}
			else 
			{
				return 0; //不匹配返回0
			}
		}	
	}
}
void Cout(int match,Stack *&p)
{
	if(match==0) cout << "NO"; //不匹配
	else 
	{
		if(p->top==-1) cout << "YES"; //判断是否栈空
		else cout << "NO";
	}
}
int main()
{
	char str[size] ;
	cin >> str;
	Stack *p;
	creat(p);
	int m=1;
	m=match(p,str);
	Cout(m,p);
	return 0;
 }