[Java] 백준 2529번 : 부등호

문제

두 종류의 부등호 기호 ‘<’와 ‘>’가 k개 나열된 순서열 A가 있다. 우리는 이 부등호 기호 앞뒤에 서로 다른 한 자릿수 숫자를 넣어서 모든 부등호 관계를 만족시키려고 한다. 예를 들어, 제시된 부등호 순서열 A가 다음과 같다고 하자. 

A ⇒ < < < > < < > < >

부등호 기호 앞뒤에 넣을 수 있는 숫자는 0부터 9까지의 정수이며 선택된 숫자는 모두 달라야 한다. 아래는 부등호 순서열 A를 만족시키는 한 예이다. 

3 < 4 < 5 < 6 > 1 < 2 < 8 > 7 < 9 > 0

이 상황에서 부등호 기호를 제거한 뒤, 숫자를 모두 붙이면 하나의 수를 만들 수 있는데 이 수를 주어진 부등호 관계를 만족시키는 정수라고 한다. 그런데 주어진 부등호 관계를 만족하는 정수는 하나 이상 존재한다. 예를 들어 3456128790 뿐만 아니라 5689023174도 아래와 같이 부등호 관계 A를 만족시킨다. 

5 < 6 < 8 < 9 > 0 < 2 < 3 > 1 < 7 > 4

여러분은 제시된 k개의 부등호 순서를 만족하는 (k+1)자리의 정수 중에서 최댓값과 최솟값을 찾아야 한다. 앞서 설명한 대로 각 부등호의 앞뒤에 들어가는 숫자는 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 }중에서 선택해야 하며 선택된 숫자는 모두 달라야 한다. 

입력

첫 줄에 부등호 문자의 개수를 나타내는 정수 k가 주어진다. 그 다음 줄에는 k개의 부등호 기호가 하나의 공백을 두고 한 줄에 모두 제시된다. k의 범위는 2 ≤ k ≤ 9 이다. 

출력

여러분은 제시된 부등호 관계를 만족하는 k+1 자리의 최대, 최소 정수를 첫째 줄과 둘째 줄에 각각 출력해야 한다. 단 아래 예(1)과 같이 첫 자리가 0인 경우도 정수에 포함되어야 한다. 모든 입력에 답은 항상 존재하며 출력 정수는 하나의 문자열이 되도록 해야 한다. 

예제 입력 1

2
< >

예제 출력 1

897
021

예제 입력 2

9
> < < < > > > < <

예제 출력 2

9567843012
1023765489

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.StringTokenizer;

public class Main {

	private static int K;
	private static char [] ch;
	private static StringTokenizer st;
	private static List<String> list=new ArrayList<String>();
	public static void main(String[] args) throws NumberFormatException, IOException {

		BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
		K=Integer.parseInt(bf.readLine());
		ch=new char[K];
		st=new StringTokenizer(bf.readLine(), " ");
		for(int i=0;i<K;i++) {
			ch[i]=st.nextToken().charAt(0);
		}
		
		permutation(K+1, new int[K+1], new boolean[10]);
		Collections.sort(list);
		System.out.println(list.get(list.size()-1));	// 가능한 모든 경우의 수 중에서 최대값인 경우
		System.out.println(list.get(0));	// 가능한 모든 경우의 수 중에서 최솟값인 경우
	}

	public static void permutation(int toChoose, int [] choosed, boolean [] visited) {	// K+1개의 수로 만들 수 있는 모든 순열의 경우의 수
		if(toChoose==0) {	// 더 이상 뽑을 개수가 없는 경우, 부등호를 확인하여 가능한 경우를 list에 추가한다.
			for(int i=0;i<K;i++) {
				int start=choosed[i];
				int second=choosed[i+1];
				if(ch[i]=='<') {
					if(start>second) return;
				}
				if(ch[i]=='>') {
					if(start<second) return;
				}
			}
			String s="";
			for(int r:choosed) {
				if(r==0) s+="0";
				else {
					String tmp=Integer.toString(r);
					s+=tmp;
				}
			}
			list.add(s);
			return;
		}
		
		for(int i=0;i<10;i++) {
			if(!visited[i]) {
				visited[i]=true;
				choosed[choosed.length-toChoose]=i;
				permutation(toChoose-1, choosed, visited);
				visited[i]=false;
			}
		}
	}
}

1. 완전 탐색을 통해 가능한 모든 순열을 만들어줍니다.

2. 그 중에서 부등호를 만족하는 경우를 전부 리스트에 추가합니다.

3. 리스트를 정렬합니다.

4. 정렬된 리스트에서 최대값과 최솟값을 출력합니다.