[프로그래머스] 괄호 변환(C++)

문제 URL

https://school.programmers.co.kr/learn/courses/30/lessons/60058

프로그래머스

[문제 설명]

"균형 잡힌 괄호 문자열" p가 매개변수로 주어질 때, 주어진 알고리즘을 수행해 "올바른 괄호 문자열"로 변환한 결과를 return 하는 문제이다.

올바른 괄호 문자열로 변환하는 방식은 아래와 같다.

 

1. 입력이 빈 문자열인 경우, 빈 문자열을 반환합니다. 
2. 문자열 w를 두 "균형잡힌 괄호 문자열" u, v로 분리합니다. 단, u는 "균형잡힌 괄호 문자열"로 더 이상 분리할 수 없어야 하며, v는 빈 문자열이 될 수 있습니다. 
3. 문자열 u가 "올바른 괄호 문자열" 이라면 문자열 v에 대해 1단계부터 다시 수행합니다. 
  3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다. 
4. 문자열 u가 "올바른 괄호 문자열"이 아니라면 아래 과정을 수행합니다. 
  4-1. 빈 문자열에 첫 번째 문자로 '('를 붙입니다. 
  4-2. 문자열 v에 대해 1단계부터 재귀적으로 수행한 결과 문자열을 이어 붙입니다. 
  4-3. ')'를 다시 붙입니다. 
  4-4. u의 첫 번째와 마지막 문자를 제거하고, 나머지 문자열의 괄호 방향을 뒤집어서 뒤에 붙입니다. 
  4-5. 생성된 문자열을 반환합니다.

 

 

[문제 풀이]

올바른 괄호 문자열로 변환하는 방식을 그대로 구현하면 되는 문제이다. 

3단계와 4단계를 구분하기 위해 문자열 u가 올바른 괄호 문자인지 판단하는 코드가 필요하다.

---

#include <string>
#include <vector>
#include <stack>
using namespace std;

string solution(string p) {
    string answer = "";

    // 1단계
    if(p == "")
        return p;
    
    // 2단계
    int lcnt = 0, rcnt = 0;
    string u, v;
    for(int i=0; i<p.length(); i++) {
        u += p[i];
        if(p[i] == '(')
            lcnt++;
        else
            rcnt++;
        
        if(lcnt==rcnt) {
            for(int j=i+1; j<p.length(); j++)
                v+=p[j];
            break;
        }
    }
    
    // 3단계-올바른 괄호 판단
    stack<char> s;
    bool flag = true;  // 올바른 괄호인지 판단
    for(int i=0; i<u.length(); i++) {
        if(u[i] == '(')
            s.push(u[i]);
        else {
            if(!s.empty())
                s.pop();
            else {  // 올바른 괄호가 아닐 경우
                flag = false;
                break;
            }
        }
    }
    
    // 3단계
    if(flag && s.empty()) {
        u += solution(v);
        return u;
    }
    else {  // 4단계 
        string str = "(";
        str += solution(v);
        str += ")";
        u.erase(u.begin());
        u.erase(u.end()-1);
        for(int i=0;i<u.length();i++) {
            if(u[i] == '(')
                str += ")";
            else
                str += "(";
        }
        return str;
    }
    
    return answer;
}

참고: https://non-stop.tistory.com/261