洛谷 P1241 括号序列（stack）

P1241 括号序列 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

定义如下规则：

1. 空串是「平衡括号序列」

2. 若字符串 S 是「平衡括号序列」，那么 [S] 和 (S) 也都是「平衡括号序列」

3. 若字符串 A 和 B 都是「平衡括号序列」，那么 AB（两字符串拼接起来）也是「平衡括号序列」。

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

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

而以下几个则不是：

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

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

4. 从左到右扫描整个字符串。

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

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

输入格式

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

输出格式

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

先提供一个好一点的测试数据：

输入：

([)]

输出：

()[()]

先思考！

---

思路：

字符串中的每一个字符都有俩种状态：配对成功 || 配对失败。输出时，这俩种状态的输出情况是一样的，所以需要一个vis数组标记配对成功的字符的下标。

“考察它与它左侧离它最近的未匹配的的左括号”这一点就可以想到，把左括号依放入一个栈中，遍历过程中遇到右括号，就可以在栈顶找到最近的左括号，这样就不用在以该右括号为起点向右遍历，大大缩小复杂度。

遍历到右括号时，和栈顶进行判断，如果成功配对就在vis中标记，并把配对成功的左括号弹出栈，否则就不管。

输出的时候，有vis标记的直接输出，否则就补充输出。

注意：“把左括号依放入一个栈中”是左括号的下标放入栈中，因为vis标记的就是下标。

#include<bits/stdc++.h> 
using namespace std;
string s;
stack<int> st; 
bool vis[1002];//vis标记过的 都是配对成功的  
int main()
{
    cin>>s;
    int len=s.length();
    for(int i=0;i<len;i++){
        if(s[i]=='['||s[i]=='('){
            //左括号入栈 栈st保存所有左括号的下标 
            //因为要找右括号 左边 出现的 第一个 左括号
            //所以右括号是不入栈的 
            st.push(i); 
        }
        else{
            //右括号
            //判断栈顶（即左边出现的第一个左括号的下标）
            if(s[i]==']'){
                if(st.empty()) continue;
                int k=st.top();
                if(s[k]=='['){
                    //配对成功 
                    vis[i]=vis[k]=true;
                    st.pop(); 
                }
            } 
            if(s[i]==')'){
                if(st.empty()) continue;
                //stack在读取top或者pop之前都必须要用empty判断一次
                //否则很可能re
                int k=st.top();
                if(s[k]=='('){
                    //配对成功 
                    vis[i]=vis[k]=true;
                    st.pop(); 
                }
            } 
        }
    }
    //输出处理 
    for(int i=0;i<len;i++){
        if(vis[i]) cout<<s[i];
        else{
            if(s[i]=='('||s[i]==')') cout<<"()";
            if(s[i]==']'||s[i]=='[') cout<<"[]";
        }
    } 
    return 0;
}

由此可以看出：栈在题目中使用的时候，大多是作为一个中转的处理工具，而不是存放输出答案的容器，当然只是一部分。