P1241 括号序列【stack】

括号序列

题目描述

定义如下规则：

1. 空串是「平衡括号序列」
2. 若字符串 $S$ 是「平衡括号序列」，那么 $\text{[}S\text{]}$ 和 $\text{(}S\text{)}$ 也都是「平衡括号序列」
3. 若字符串 $A$ 和 $B$ 都是「平衡括号序列」，那么 $AB$（两字符串拼接起来）也是「平衡括号序列」。

例如，下面的字符串都是平衡括号序列：

• ()，[]，(())，([])，()[]，()[()]

而以下几个则不是：

• (，[，]，)(，())，([()

现在，给定一个仅由 (，)，[，]构成的字符串 $s$，请你按照如下的方式给字符串中每个字符配对：

1. 从左到右扫描整个字符串。
2. 对于当前的字符，如果它是一个右括号，考察它与它左侧离它最近的未匹配的的左括号。如果该括号与之对应（即小括号匹配小括号，中括号匹配中括号），则将二者配对。如果左侧未匹配的左括号不存在或与之不对应，则其配对失败。

配对结束后，对于 $s$ 中全部未配对的括号，请你在其旁边添加一个字符，使得该括号和新加的括号匹配。

输入格式

输入只有一行一个字符串，表示 $s$。

输出格式

输出一行一个字符串表示你的答案。

样例 #1

样例输入 #1

([()

样例输出 #1

()[]()

样例 #2

样例输入 #2

([)

样例输出 #2

()[]()

提示

数据规模与约定

对于全部的测试点，保证 $s$ 的长度不超过 $100$，且只含 (，)，[，] 四种字符。

问题链接： P1241 括号序列
问题分析： 括号匹配问题，用堆栈来解决。
参考链接： （略）
题记： （略）

AC的C++语言程序如下：

/* P1241 括号序列 */

#include <iostream>
#include <stack>
#include <cstring>

using namespace std;

const int N = 100 + 1;
bool flag[N];

int main()
{
    string s;
    cin >> s;
    memset(flag, false, sizeof flag);
    stack<int> stk;
    for (int i = 0; s[i]; i++) {
        if (s[i] == ')')  {
            if (stk.empty()) continue;
            if (s[stk.top()] == '(') {
                flag[i] = flag[stk.top()] = true;
                stk.pop();
            }
        } else if (s[i] == ']') {
            if (stk.empty()) continue;
            if (s[stk.top()] == '[') {
                flag[i] = flag[stk.top()] = true;
                stk.pop();
            }
        } else
            stk.push(i);
    }
    for (int i = 0; s[i]; i++) {
        if (flag[i])
            cout << s[i];
        else {
            if (s[i] == '(' || s[i] == ')')
                cout << "()";
            else
                cout << "[]";
        }
    }
    return 0;
}

AC的C++语言程序如下：

/* P1241 括号序列 */

#include <iostream>

using namespace std;

const int N = 100 + 1;
int stk[N], top;
char a[N];

int main()
{
    string s;
    cin >> s;
    top = 0;
    for (int i = 0; s[i]; i++) {
        if (s[i] == '(') 
            stk[++top] = i, a[i] = ')';
        else if (s[i] == '[')
            stk[++top] = i, a[i] = ']';
        else if (s[i] == ')') {
            if (top == 0 || a[stk[top]] != s[i])
                a[i] = '(';
            else
                a[stk[top--]] = ' ';
        } else if (s[i] == ']') {
            if (top == 0 || a[stk[top]] != s[i])
                a[i] = '[';
            else
                a[stk[top--]] = ' ';
        }
    }
    for (int i = 0; s[i]; i++) {
        if (a[i] == '(' || a[i] == '[')
            cout << a[i];
        cout << s[i];
        if (a[i] == ')' || a[i] == ']')
            cout << a[i];
    }
    return 0;
}