쵼쥬 2021. 10. 6. 16:30


풀이 방법

각 괄호의 경우의 수를 구하기 위해 조합을 뽑아냈다.

먼저 괄호의 인덱스를 map에 저장해주고 그 인덱스르 바탕으로 조합의 경우의 수를 계산했다.

경우의 수를 바탕으로 괄호를 제거하고 출력해 주었다.

재귀 함수로 조합을 계산할 수 있다.

(((0))) <- 이런 테스트 케이스는 중복을 고려해야 되기 때문에 set으로 중복을 제거해주었다.

 

내 코드

package com.company;

import java.io.*;
import java.util.*;


public class Main {
    static String s = "";
    static HashMap<Integer, Integer> map;
    static Stack<Integer> stack;
    static HashSet<String> result;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        s = br.readLine();
        map = new HashMap<>();
        stack = new Stack<>();
        result = new HashSet<>();
        ArrayList<Integer> key = new ArrayList<>();

        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == '(') {
                stack.push(i);
            } else if (s.charAt(i) == ')') {
                map.put(stack.pop(), i);
            }
        }

        for (Integer integer : map.keySet()) {
            key.add(integer);
        }

        boolean[] visited = new boolean[key.size()];

        for (int i = 1; i <= map.size(); i++) {
            combination(key, visited, 0, i);
        }

        ArrayList<String> resultList = new ArrayList(result);
        Collections.sort(resultList);
        for (String s1 : resultList) {
            System.out.println(s1);
        }
    }

    static void combination(ArrayList<Integer> key, boolean[] visited, int start, int r) {
        if (r == 0) {

            print(s, map, key, visited);

            return;
        }

        for (int i = start; i < key.size(); i++) {
            if (!visited[i]) {
                visited[i] = true;
                combination(key, visited, i + 1, r - 1);
                visited[i] = false;
            }
        }
    }


    static void print(String s, HashMap<Integer, Integer> map, ArrayList<Integer> key, boolean[] visited) {
        StringBuilder sb = new StringBuilder(s);
        for (int i = 0; i < key.size(); i++) {
            if (visited[i]) {
                sb.setCharAt(key.get(i), '!');
                sb.setCharAt(map.get(key.get(i)), '!');
            }
        }

        s = sb.toString();
        result.add(s.replaceAll("!", ""));
    }
}