[BOJ] 2529 부등호 - JAVA

1. 문제

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 }중에서 선택해야 하며 선택된 숫자는 모두 달라야 한다. 

---

2. 풀이

K	부등호 개수
int[] pick	뽑은 숫자 저장한 배열
boolean[] number	뽑을 수 있는 숫자(0~9)
boolean[] input	입력받은 부등호. <이면 true, >이면 false를 저장했다.
ArryaList<String> list	가능한 문자열을 모두 저장한 리스트

1.  문제 풀이는 순열을 이용해서 가능한 경우를 모두 list에 저장하고 정렬한 뒤 0번째 원소와 마지막 원소를 출력했다.

2.  cnt가 K+1이면 종료하고 pick에 있는 숫자를 StringBuilder에 더한 뒤 String으로 만들어서 list에 넣는다.

3.  cnt가 K+1보다 작은 경우 i를 0부터 9까지 돌면서 탐색한다. 이미 뽑은 숫자가 아니면서 이전에 뽑은 숫자(pick[cnt-1])와 지금 뽑으려는 숫자(i)의 크기를 비교해서 입력받은 부등호와 동일하면 i를 뽑을 수 있다.

4.  3번의 경우 number[i]를 true처리한 뒤 pick[cnt]에 i를 저장한다. 그 뒤 재귀로 Permutation함수를 부르고, 전부 부른 뒤에는 number[i]를 false로 되돌려놓는다. pick[cnt]는 어차피 다른 값으로 덮일 것이므로 돌리지 않아도 된다.

import java.io.*;
import java.util.*;

public class BOJ_2529_부등호 {
   static int K;
   static int[] pick;
   static boolean[] number = new boolean[10];
   static boolean[] input;
   static ArrayList<String> list;
   public static void main(String[] args) throws IOException {
      BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
      K = Integer.parseInt(in.readLine());
      input = new boolean[K];
      StringTokenizer st = new StringTokenizer(in.readLine());
      for(int i=0;i<K;i++) input[i] = st.nextToken().charAt(0) == '<';
      pick = new int[K+1];

      list = new ArrayList<>();
      Permutation(0);
      Collections.sort(list);
      System.out.println(list.get(list.size()-1)+"\n"+list.get(0));
   }
   private static void Permutation(int cnt){
      if(cnt==K+1){
         StringBuilder sb = new StringBuilder();
         for(int i=0;i<=K;i++) sb.append(pick[i]);
         list.add(sb.toString());
         return;
      }

      for(int i=0;i<10;i++){
         if(number[i] || (cnt>0 && pick[cnt-1]<i != input[cnt-1])) continue;
         number[i] = true;
         pick[cnt] = i;
         Permutation(cnt+1);
         number[i] = false;
      }
   }
}

---

3. 결과