小马智行9.20 第四题

#include<iostream>
#include<vector>
#include<string>
#include<unordered_map>
#include<stack>

using namespace std;

int main() {
	int n;
	cin >> n;
	vector<string> s;
	for (int i = 0; i < n; i++) {
		string tem;
		cin >> tem;
		s.push_back(tem);
	}
	unordered_map<char,char> m;
	m[')'] = '(';
	m[']'] = '[';

	for (int i = 0; i < n; i++) {
		string t = s[i];
		int len = t.length();
		stack<char> c;
		stack<char> k;
		int  flag = 1;
		for (int j = 0; j < len; j++) {
			if ((t[0] != '(') || (t[j] == '[' && j < len - 1 && t[j + 1] != '(') || (t[j] == ',' && j < len - 1 && t[j + 1] != '(')) {
				flag = -1;
				break;
			}
			if (t[j] == ',')
				c.push(t[j]);

			if (c.size() > 1) {
				flag = -1;
				break;
			}
			if (t[j] == '[' || t[j] == ']')
			{
				while (!c.empty())
					c.pop();
			}
			if (t[j] == '(' || t[j] == '[')
				k.push(t[j]);
			if (t[j] == ')' || t[j] == ']') {
				if (k.top() == m[t[j]])
					k.pop();
			}
		}
		if (!k.empty()) {
			flag = -1;
			while (!k.empty())
				k.pop();
		}
		if (flag > 0)
			cout << 1 << '\n';
		else
			cout << 0 << '\n';
	}
	return 0;
}