[백준] 리모컨 - Java

문제 링크

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

문제

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

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

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

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

입력

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

출력

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

예제 입력 1 

5457
3
6 7 8

예제 출력 1 

6

예제 입력 2 

100
5
0 1 2 3 4

예제 출력 2 

0

예제 입력 3

500000
8
0 2 3 4 6 7 8 9

예제 출력 3

11117

예제 입력 4

100
3
1 0 5

예제 출력 4

0

예제 입력 5

14124
0

예제 출력 5

5

예제 입력 6

1
9
1 2 3 4 5 6 7 8 9

예제 출력 6

2

예제 입력 7

80000
2
8 9

예제 출력 7

2228

풀이 방법

• 고장난 버튼을 set에 담아 시간복잡도를 O(1)로 설정하였다.
• 또한, 0번부터 999999번까지(50만 채널까지 있으므로, 100만 이상은 넘어가지 않을 것) 채널을 반복하여 기준을 잡고, 부서지지 않은 번호들로 구성된 번호에서 업다운을 눌러서 해당 채널까지 가는 최소의 값을 구해주었다.

• 고장난 버튼을 set에 넣어, 반복문에서 check를 먼저 수행하고 채널에 누르지 못하는 번호가 있는지 확인하고, true일 때만 수행하게 된다.
• 이 때, M이 0인경우(최악의 경우)에도 100만번 반복하므로 시간상에 문제는 없다고 판단했다.

풀이 코드

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

public class Main{
    static int N, M;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        // 이동하려는 채널
        N = Integer.parseInt(br.readLine());

        // 고장난 버튼이 있을경우
        M = Integer.parseInt(br.readLine());
        Set<Integer> broken = new HashSet<>();

        // 고장난 버튼이 있는 경우에만 수행
        if (M > 0) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            // 값 추가
            for (int i = 0; i < M; i++) {
                broken.add(Integer.parseInt(st.nextToken()));
            }
        }

        // 현재 채널이면 바로 끝
        if (N == 100) {
            System.out.println(0);
            return;
        } else {
            System.out.println(minClicks(100, N, broken));
        }
    }

    public static int minClicks(int current, int target, Set<Integer> broken) {

        // 업 다운만 하는 경우.
        int minClicks = Math.abs(target - current);

        // 숫자를 눌러 이동하는 경우
        for (int channel = 0; channel <= 999999; channel++) {
            if (checkBrokenButton(channel, broken)) {
                int clicks = String.valueOf(channel).length() + Math.abs(channel - target);
                minClicks = Math.min(minClicks, clicks);
            }
        }

        return minClicks;
    }

    // 부서진 번호 없이도 입력가능한지 확인
    public static boolean checkBrokenButton(int channel, Set<Integer> broken) {
        if (channel == 0) {
            return !broken.contains(0);
        }

        while (channel > 0) {
            int digit = channel % 10;
            if (broken.contains(digit)) {
                return false;
            }
            channel /= 10;
        }

        return true;
    }
}

결과

회고

• 브루트포스 알고리즘은 생각할 때 효율성과 정확성 모두 따져야해서 생각이 많이 들어가는 문제인 것 같다...
• 익숙해지자