유블로그

BOJ 1759 암호만들기 자바 java 본문

알고리즘

BOJ 1759 암호만들기 자바 java

yujeong kang 2021. 6. 19. 13:32

BOJ 1759 암호만들기 자바 java

 

https://www.acmicpc.net/problem/1759

 

1759번: 암호 만들기

첫째 줄에 두 정수 L, C가 주어진다. (3 ≤ L ≤ C ≤ 15) 다음 줄에는 C개의 문자들이 공백으로 구분되어 주어진다. 주어지는 문자들은 알파벳 소문자이며, 중복되는 것은 없다.

www.acmicpc.net

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
	static int L, C;
	static char[] input;
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		L = Integer.parseInt(st.nextToken());
		C = Integer.parseInt(st.nextToken());
		arr = new char[L];
		input = new char[C];
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < C; i++) {
			input[i] = st.nextToken().charAt(0);
		}
		Arrays.sort(input);
		solve(0, 0, 0, 0);
		
	} // main
	
	static char[] arr;
	private static void solve(int start, int cnt, int vCnt, int cCnt) {	// vCnt : 모음개수, cCnt : 자음개수
		if(cnt == L) {
			if(vCnt < 1 || cCnt < 2) return;
			StringBuilder sb = new StringBuilder();
			for (int j = 0; j < L; j++) {
				sb.append(arr[j]);
			}
			System.out.println(sb.toString());
			return;
		}
		for (int i = start; i < C; i++) {
			arr[cnt] = input[i];
			boolean isMo = isMo(input[i]);
			solve(i+1, cnt+1, isMo? vCnt+1 : vCnt, isMo? cCnt : cCnt+1);
		}
	}
	
	static boolean isMo(char c) {
		if(c == 'a' || c == 'e' || c == 'i' || c == 'o' || c == 'u') {
			return true;
		}
		return false;
	}
}

 

 

조합 문제인데 약간의 백트래킹..?

그냥 C개 중 L개 다 뽑아서 그 때마다 자음 모음 체크하여 출력할 수도 있지만

나는 그냥 함수 인자에 자음 개수와 모음 개수를 넣어

L개를 다 뽑았을 때 모음이 1개 보다 적거나 자음이 2개 보다 적으면 출력하지 않고 그냥 return 했다.