[백준] 1107

리모컨

문제

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

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

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

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

입력

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

출력

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

풀이

#include <iostream>
#include <string>

bool disabled[10];

bool numbers[1000000];

void check_number(int i) {
    std::string str = std::to_string(i);
    int length = str.length();
    for (int j = 0; j < length; j++) {
        if (disabled[str[j] - '0']) numbers[i] = true;
    }
}

int get_digit(int i) {
    std::string str = std::to_string(i);
    int length = str.length();
    return length;
}

int main(void) {
    int n, m, temp, ans, low_digit, upp_digit;
    int lower = -1;
    std::cin >> n >> m;
    int upper = 1000000;

    for (int i = 0; i < m; i++) {
        std::cin >> temp;
        disabled[temp] = true;
    }

    for (int i = 0; i <= 1000000; i++) {
        check_number(i);
        if (i > lower && i <= n && !numbers[i]) lower = i;
        if (i > n && !numbers[i]) {
            upper = i;
            break;
        }
    }

    low_digit = get_digit(lower);
    upp_digit = get_digit(upper);
    if (lower < 0 || n - lower + low_digit > upper - n + upp_digit) {
        ans = upper - n + upp_digit;
    } else {
        ans = n - lower + low_digit;
    }
    ans = std::min(std::abs(100 - n), ans);
    std::cout << ans << "\n";
}

반성회

브루트포스로 100만개의 경우의 수를 전부 구해서 풀었다

최대한 100만단위 for문을 자주 돌지 않도록 하나의 for문에 모든 동작을 다 하도록 했다