[BOJ_2529_JAVA] 부등호

 

문제

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

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인 경우도 정수에 포함되어야 한다. 모든 입력에 답은 항상 존재하며 출력 정수는 하나의 문자열이 되도록 해야 한다. 

 

문제풀이

순서대로 입력을 받고, DFS를 통해 부등호 모양과 이미 사용한 숫자인지 판별만 해준다면 쉽게 풀 수 있다.

다만, 선택된 숫자가 중복이 되지 않으므로 조합으로 풀어야하고 예제를 보면 답이 021인데, 앞에 0이라는 숫자는 Long 타입으로 준다면 사라지기 때문에 min 값을 String 형식으로 기억해야한다.

 

소스코드

package com.ssafy.algorithm.boj_2529_부등호;

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

public class Main {
	public static boolean[] visited = new boolean[10];
	public static char[] arr;
	public static int[] nums;
	public static int K;
	public static Long max = 0L, min = Long.MAX_VALUE;
	public static String sMax, sMin;
	
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		K = Integer.parseInt(br.readLine());
		arr = new char[K];
		nums = new int[K+1];
		
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		for (int i = 0; i < K; i++) {
			arr[i] = st.nextToken().charAt(0);
		}
		
		for (int i = 0; i < 10; i++) {
			nums[0] = i;
			visited[i] = true; 
			dfs(1);
			visited[i] = false;
		}
		System.out.println(sMax);
		System.out.println(sMin);
	}
	
	public static void dfs(int cnt) {
		if(cnt == nums.length) {
			String tmp = "";
			for (int i = 0; i < nums.length; i++) {
				tmp += nums[i];
			}
			
			Long value = Long.parseLong(tmp);
//			max = max > value ? max : value;
			if(max < value) {
				max = value;
				sMax = tmp;
			}
//			min = min < value ? min : value;
			if(min > value) {
				min = value;
				sMin = tmp;
			}
			return;
		}
		
		for (int i = 0; i < 10; i++) {
			boolean tri = false;
			
			switch(arr[cnt-1]) {
			case '<':
				if (nums[cnt-1] < i) tri = true; 
				break;
			case '>':
				if (nums[cnt-1] > i) tri = true;
				break;
			}
			
			if(!visited[i] && tri) {
				visited[i] = true;
				nums[cnt] = i;
				dfs(cnt+1);
				visited[i] = false;
			}
		}
	}
}