백준 1107번 리모컨 (java)

[Gold V] 리모컨 - 1107

문제 링크

성능 요약

메모리: 14436 KB, 시간: 204 ms

분류

브루트포스 알고리즘

제출 일자

2024년 12월 18일 12:51:23

문제 설명

수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.

리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.

수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

수빈이가 지금 보고 있는 채널은 100번이다.

입력

첫째 줄에 수빈이가 이동하려고 하는 채널 N (0 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 고장난 버튼의 개수 M (0 ≤ M ≤ 10)이 주어진다. 고장난 버튼이 있는 경우에는 셋째 줄에는 고장난 버튼이 주어지며, 같은 버튼이 여러 번 주어지는 경우는 없다.

출력

첫째 줄에 채널 N으로 이동하기 위해 버튼을 최소 몇 번 눌러야 하는지를 출력한다.

풀이

• 두 경우로 나눌 수 있다.
  • +와 - 키만 눌러서 이동한 경우
  • 리모컨 번호를 누른 후 +와 -로 이동한 경우
• 가장 먼저 최솟값에 (이동하려는 채널 - 100) 값을 집어넣어 +와 -만 눌러서 이동한 경우를 처리해줬다.
• 이후 완전 탐색으로 리모컨 번호를 누른 후 이동하는 모든 경우의 수를 탐색했다.

주의할 점은 예외가 굉장히 많다.

예를 들어 600번으로 이동하는 경우에

2,3,4,5,6,7,8,9 번호가 금지된 상황이라고 가정해보자

 

이 경우 111번에서 이동하는 것보다 1000번에서 이동하는 게 더 빠르게 도착한다.

 

많은 케이스를 넣어보고 예외를 잘 처리해보도록 하자.

 

코드

public class Main {
    static int min = 500000, x;
    static int[] forbidden;
    static int digit;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        x = Integer.parseInt(br.readLine());

        int t = Integer.parseInt(br.readLine());
        StringTokenizer st = null;
        forbidden = new int[t];
        if(t != 0) {
            st = new StringTokenizer(br.readLine());
        }

        for(int i =0;i<t;i++){
            forbidden[i] = Integer.parseInt(st.nextToken());
        }
        digit = getDigit(x);
        min = Math.abs(x - 100);
        dfs(0,0);
        System.out.println(min);
    }

    public static void dfs(int depth, int result){
        if(depth != 0){
            min = Math.min(Math.abs(result-x) + getDigit(result), min);
        }
        if(depth == digit+1){
            return;
        }

        for(int  j=0;j<10;j++){
            if(can(j)){
                dfs(depth+1,result * 10 + j);
            }
        }

    }

    public static boolean can(int num){
        for (int i : forbidden) {
            if(num == i){
                return false;
            }
        }
        return true;
    }

    public static int getDigit(int num){
        int digit = 1;
        while(num >= 10){
            num /= 10;
            digit++;
        }
        return digit;
    }
}