P.1107 리모컨

castlehi·2022년 2월 25일
0

CodingTest

목록 보기
13/67
post-thumbnail

1107 리모컨

시간 제한메모리 제한제출정답맞힌 사람정답 비율
2 초 256 MB6117214450992122.441%

문제

수빈이는 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

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

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

        int cur = 100;
        int tar = Integer.parseInt(br.readLine());
        int n = Integer.parseInt(br.readLine());

        if (n > 0) {
            int[] btn = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
            int res;

            int min = Integer.MAX_VALUE;
            if (tar < 100) {
                while (cur >= 0) {
                    if (not_breakdown(cur, btn)) {
                        res = Math.abs(cur - tar) + ((Integer.toString(cur).length() > 100 - cur) ? 100 - cur : Integer.toString(cur).length());
                        min = (min > res) ? res : min;
                    }
                    else {
                        if (cur == tar) min = (min > 100 - cur) ? 100 - cur : min;
                    }
                    cur--;
                }
            } else if (tar == 100) min = 0;
            else {
                while (cur <= 999999) {
                    if (not_breakdown(cur, btn)) {
                        res = Math.abs(cur - tar) + ((Integer.toString(cur).length() > cur - 100) ? cur - 100 : Integer.toString(cur).length());
                        min = (min > res) ? res : min;
                    }
                    else {
                        if (cur == tar) min = (min > cur - 100) ? cur - 100 : min;
                    }
                    cur++;
                }
            }

            System.out.println(min);
        }
        else
            System.out.println((Integer.toString(tar).length() > Math.abs(tar - 100)) ? Math.abs(tar - 100) : Integer.toString(tar).length());
    }

    public static boolean not_breakdown(int cur, int[] btn) {
        int a, b;
        a = cur;

        if (a != 0) {
            while (a != 0) {
                b = a % 10;
                a /= 10;
                for (int i = 0; i < btn.length; i++) {
                    if (b == btn[i]) return false;
                }
            }
        }
        else {
            for (int i = 0; i < btn.length; i++) {
                if (a == btn[i]) return false;
            }
        }
        return true;
    }
}

코드 설명

0부터 999999까지 모두 돌리는 방법을 선택했다. 시간 제한이 2초이기 때문에 별다른 생각 없이 브루트 포스를 이용하기로 했다. 가까운 값을 선택해서 +나 -하는 방법도 있지만 시간 제한에 비해 코드가 너무 복잡해지기 때문에 그 방법은 선택하지 않았다.

큰 조건은 고장난 버튼 유무로 나누었다. 고장난 버튼이 없다면 직접 원하는 tv채널 버튼을 눌러서 바꾸거나 기준점인 100에서 +나 - 버튼을 눌러 바꾸는 방법 두 가지가 있다.

고장난 버튼이 있다면 현재 채널을 바꿔가면서 이 채널의 번호의 버튼의 고장 유무로 나눠서 구했다. 고장난 버튼이 포함되어 있다면 100에서 +나 -를 눌러 바꾸는 방법밖에 없고, 고장난 버튼이 없다면 현재 버튼을 누르기 위한 횟수와 현재 버튼과 바꾸고 싶은 채널의 차이를 더한 값이 채널을 바꾸기 위해 버튼을 누르는 개수가 된다.

이 방법을 통해 min을 업데이트 해 가면서 바꿔주었다.

만약, 바꾸고 싶은 채널이 현재 채널 (100)과 같다면 별다른 계산 없이 바로 0을 출력하도록 했다.

profile
Back-end Developer

0개의 댓글

관련 채용 정보