본문 바로가기

알고리즘/중급1

[JAVA] 백준 2529번: 부등호 ( 중급 1-1 )

문제

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

 

문제 풀이

모든 경우를 다 확인하는 브루트 포스 방법을 이용하였다.

3개의 숫자가 완성되었을 때, 부등호 조건이 일치하는지 확인하고 있다.

전체 코드를 확인 해보자.

 

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

public class baekjoon2529_4 {
    private static ArrayList<String> resultList = new ArrayList<>();
    private static int n;
    private static String[] info;
    private static boolean[] visit = new boolean[10];

    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        info = new String[n];
        int[] result = new int[n + 1];

        for (int i = 0; i < n; i++) {
            info[i] = st.nextToken();
        }

        for (int i = 0; i < 10; i++) {
            visit[i] = true;
            result[0] = i;
            bruteForce2529(1, result);
            visit[i] = false;
        }

        Collections.sort(resultList);
        String min = resultList.get(0);
        String max = resultList.get(resultList.size() - 1);

        System.out.println(max);
        System.out.println(min);
    }

    private static void bruteForce2529(int index, int[] result) {
        if (index == n + 1) {
            if (checkCondition2529(result)) {
                String temp = "";
                for (int i : result) {
                    temp += Integer.toString(i);
                }
                resultList.add(temp);
            }
            return;
        }

        for (int i = 0; i < 10; i++) {
            if (!visit[i]) {
                visit[i] = true;
                result[index] = i;
                bruteForce2529(index + 1, result);
                visit[i] = false;
            }
        }
    }

    private static boolean checkCondition2529(int[] result) {
        for (int i = 0; i < n; i++) {
            if (info[i].equals(">") && result[i] <= result[i + 1]) {
                return false;
            }
            if (info[i].equals("<") && result[i] >= result[i + 1]) {
                return false;
            }
        }
        return true;
    }
}

 

매 step 마다 부등호 조건이 맞는지 확인하는 방법을 이용해 코드를 작성하였다.

 

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

public class baekjoon2529_5 {
    private static ArrayList<String> resultList = new ArrayList<>();
    private static int n;
    private static String[] info;
    private static boolean[] visit = new boolean[10];

    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        info = new String[n];
        int[] result = new int[n + 1];

        for (int i = 0; i < n; i++) {
            info[i] = st.nextToken();
        }

        for (int i = 0; i < 10; i++) {
            visit[i] = true;
            result[0] = i;
            bruteForce2529(1, result);
            visit[i] = false;
        }

        Collections.sort(resultList);
        String min = resultList.get(0);
        String max = resultList.get(resultList.size() - 1);

        System.out.println(max);
        System.out.println(min);
    }

    private static void bruteForce2529(int index, int[] result) {
        if (index == n + 1) {
            String temp = "";
            for (int i : result) {
                temp += Integer.toString(i);
            }
            resultList.add(temp);
            return;
        }

        for (int i = 0; i < 10; i++) {
            result[index] = i;
            if (!visit[i] && checkCodition2529_4(index, result)) {
                visit[i] = true;
                bruteForce2529(index + 1, result);
                visit[i] = false;
            }
        }
    }

    private static boolean checkCodition2529_4(int index, int[] result) {
        if (info[index - 1].equals(">") && result[index - 1] > result[index]) {
            return true;
        }
        if (info[index - 1].equals("<") && result[index - 1] < result[index]) {
            return true;
        }
        return false;
    }
}