알고리즘
[백준] 리모컨 - Java
BGK97
2025. 1. 15. 17:13
수빈이는 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;
}
}결과
회고
- 브루트포스 알고리즘은 생각할 때 효율성과 정확성 모두 따져야해서 생각이 많이 들어가는 문제인 것 같다...
- 익숙해지자