PS/백준
백준 - 문자열 집합 조합하기 25328
팡세영
2023. 3. 3. 23:34
[문제 해결]
- 문자열 X, Y, Z 에서 K개를 선택해 나온 조합에서 같은 문자열이 2개 이상 나온 문자열을 출력하는 문제
- X, Y, Z 조합을 구하면서 HashMap을 이용해 문자열이 나온 빈도수를 기록한다.
- 빈도수가 2 이상인 문자열들을 오름차순 출력
[Java]
import java.io.*;
import java.util.*;
public class Main {
static HashMap<String, Integer> map = new HashMap<>();
public static void main(String args[]) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String x = br.readLine();
String y = br.readLine();
String z = br.readLine();
char xArr[] = x.toCharArray();
char yArr[] = y.toCharArray();
char zArr[] = z.toCharArray();
int k = Integer.parseInt(br.readLine());
combination(xArr, new boolean[xArr.length], 0, k);
combination(yArr, new boolean[yArr.length], 0, k);
combination(zArr, new boolean[zArr.length], 0, k);
ArrayList<String> list = new ArrayList<>();
for(String key : map.keySet()){
if(map.get(key) >= 2){
list.add(key);
}
}
if(list.size() == 0){
System.out.println(-1);
return;
}
Collections.sort(list);
for(String ans : list) System.out.println(ans);
}
public static void combination(char arr[], boolean visited[], int start, int r){
if(r == 0){
StringBuilder sb = new StringBuilder("");
for(int i=0; i<arr.length; i++)
if(visited[i]) sb.append(arr[i]);
String key = sb.toString();
map.put(key, map.getOrDefault(key, 0) + 1);
return;
}
for(int i=start; i<arr.length; i++){
if(!visited[i]){
visited[i] = true;
combination(arr, visited, i + 1, r - 1);
visited[i] = false;
}
}
}
}