[프로그래머스] 괄호 변환 (2020 KAKAO BLIND RECRUITMENT) - Java

도대체 무슨 말인지 이해가 안됐는데 “뇌 빼고 풀어라”는 글이 많길래 글자 그대로 하라는대로 구현하니까 풀렸다.

문제

괄호 변환 (2020 KAKAO BLIND RECRUITMENT)

아이디어

재귀에 관한 문제였다.

문제를 보면 이런 설명이 있는데,

문자열 w가 “균형잡힌 괄호 문자열” 이라면 다음과 같은 과정을 통해 “올바른 괄호 문자열”로 변환할 수 있습니다.

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. 생성된 문자열을 반환합니다.

정말 이런 로직을 가진 재귀 함수를 만들면 된다.

2번 줄에서 u, v로 분리하는 로직은 다음과 같다.

1. 문자열 w의 앞에서부터 (와 )의 개수를 센다.
2. 그 개수가 같아지는 순간 균형잡힌 괄호 문자열이 되므로, 이 때의 인덱스를 기준으로 u, v로 분리한다.
   그리고 탐색을 멈춘다.

String u = "", v = "";
int l = 0, r = 0;
for (int i=0; i<w.length(); i++) {
    if (w.charAt(i) == '(') {
        l++;
    } else {
        r++;
    }
    if (l == r) {
        u = w.substring(0, i+1);
        v = w.substring(i+1);
        break;
    }
}

3번 줄의 올바른 괄호 문자열은 스택을 사용해 isRight() 함수로 구현했다.

public boolean isRight(String s) {
    Stack<Character> st = new Stack<>();
    for (int i=0; i<s.length(); i++) {
        if (s.charAt(i) == '(') {
            st.push(s.charAt(i));
            continue;
        }
        if (!st.isEmpty()) {
            st.pop();
        }
    }
    return st.isEmpty();
}

isRight(u)가 true라면, v를 다시 재귀 함수의 파라미터로 전달해야한다.
‘3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다’ 라고 했으므로 이 때의 리턴 값은 u + 재귀함수(v)가 된다.

그렇지 않고 isRight(u)가 false라면, 역시 v를 파라미터로 넘기면서 재귀 호출이 필요하지만 리턴 값만 달라진다.
4-1 ~ 4-4를 순서대로 붙여서, "(" + 재귀함수(v) + ")" + u의 맨앞과 맨뒤를 자르고 괄호 방향 뒤집음를 반환한다.

if (isRight(u)) {
    return u + convert(v);
}
return "(" + convert(v) + ")" + reverse(u.substring(1, u.length()-1));

전체 코드

import java.util.*;

class Solution {
    public String convert(String w) {
        if ("".equals(w)) return "";
        
        String u = "", v = "";
        int l = 0, r = 0;
        for (int i=0; i<w.length(); i++) {
            if (w.charAt(i) == '(') {
                l++;
            } else {
                r++;
            }
            if (l == r) {
                u = w.substring(0, i+1);
                v = w.substring(i+1);
                break;
            }
        }
        
        if (isRight(u)) {
            return u + convert(v);
        }
        return "(" + convert(v) + ")" + reverse(u.substring(1, u.length()-1));
    }
    
    public boolean isRight(String s) {
        Stack<Character> st = new Stack<>();
        for (int i=0; i<s.length(); i++) {
            if (s.charAt(i) == '(') {
                st.push(s.charAt(i));
                continue;
            }
            if (!st.isEmpty()) {
                st.pop();
            }
        }
        return st.isEmpty();
    }
    
    public String reverse(String s) {
        String res = "";
        for (int i=0; i<s.length(); i++) {
            if (s.charAt(i) == '(') {
                res += ')';
            } else {
                res += '(';
            }
        }
        return res;
    }
    
    public String solution(String p) {
        if (isRight(p)) {
            return p;
        }
        return convert(p);
    }
}