Algorithm

[BOJ] 리모컨 1107번 C++

따봉치치 2024. 11. 27. 14:22
728x90

 

 

문제

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

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

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

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

입력

 

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

출력

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

 

 

접근 방식

 

채널으로 이동하기 위한 두 가지 방식이 있다.

 

1. + 혹은 - 버튼으로만 채널을 이동하는 방식

2. 현재 사용가능한 버튼으로 이동 후 필요 시 +/- 버튼으로 이동하는 방식

 

2가지 방식 중 더 적게 버튼을 누르는 방식을 선택하면 된다.

 

1번의 경우 target인 채널 번호를 현재 채널인 100으로 빼면 된다.

2번의 경우 만들 수 있는 채널 번호인지, 즉 채널에 들어가는 숫자 버튼이 고장났는지 확인 후 버튼을 누르는 횟수를 계산하면 된다

 

코드

#include<bits/stdc++.h>
using namespace std;


bool isBroken[10];

bool canMakeChannel(int channel) {
    if(channel == 0) return !isBroken[0];

    while(channel > 0) {
        if(isBroken[channel % 10]) {
            return false;
        }
        channel /= 10;
    }
    return true;
}

int getChannelLength(int channel) {
    if(channel == 0) return 1;
    int len = 0;
    while(channel > 0) {
        len++;
        channel /= 10;
    }
    return len;
}

int main() {
    ios::sync_with_stdio(false); cin.tie(0);

    int num; cin>>num;
    int m; cin>>m;
    
    memset(isBroken, false, sizeof(isBroken));

    for(int i=0; i<m; i++) {
        int x; cin>>x;
        isBroken[x] = true;
    }

    int res = abs(num - 100);

    for(int i=0; i<=1000000; i++) {
        if(canMakeChannel(i)) {
            int len = getChannelLength(i);
            int pressCount = len + abs(num - i);
            res = min(res, pressCount);
        }
    }

    cout<<res;
}
728x90