풀이 방법
각 괄호의 경우의 수를 구하기 위해 조합을 뽑아냈다.
먼저 괄호의 인덱스를 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("!", ""));
}
}
'코딩테스트 > [백준] 코딩테스트 연습' 카테고리의 다른 글
짐 챙기는 숌 (0) | 2021.10.07 |
---|---|
돌다리 건너기 - 2602번 (0) | 2021.10.06 |
듣보잡 - 1764번 (0) | 2021.10.06 |
평범한 배낭 - 12865번 (0) | 2021.10.05 |
징검다리 건너기 - 21317번 (0) | 2021.10.05 |