[백준 1106] 호텔
[백준 1106] 호텔
▪︎ 문제
▪︎ 알고리즘 설계
- 예시
- 목표 고객 수 C = 10
- 도시 A: 고객 3명 확보, 비용 2
- 도시 B: 고객 5명 확보, 비용 4
- 초기값: dp[0] = 0, dp[나머지] = INF (매우 큰 값)
- dp 배열은 확보할 수 있는 고객 수를 인덱스로 하여 그에 대한 최소 비용을 배열로 나타냄
- DP 흐름:
- i=1,2: 어떤 도시도 사용 불가, 도시 A(3명)와 B(5명) 둘 다 고객 수가 i보다 큼
- i=3:
- 도시 A:
dp[3] = 2
- 도시 B:
i - 5 < 0
→ 사용 불가
- 도시 A:
- i=4: 갱신 불가
- 도시 A:
i - 3 = 1
→ dp[1] + 2 = INF + 2 - 도시 B:
i - 5 < 0
→ 사용 불가
- 도시 A:
- i=5:
- 도시 A:
i - 3 = 2
→ dp[2] + 2 = INF + 2 - 도시 B:
dp[5] = 4
- 도시 A:
- i=6:
- 도시 A:
i - 3 = 3
→ dp[3] + 2 = 2 + 2 =dp[6] = 4
- 도시 B:
i - 5 = 1
→ dp[1] + 4 = INF + 4
- 도시 A:
- i=7:
- 도시 A:
i - 3 - 3 = 1
→ dp[1] + 4 = INF + 4 - 도시 B:
i - 5 = 2
→ dp[2] + 4 = INF + 4
- 도시 A:
- i=8:
- 도시 A:
i - 3 = 5
→ dp[5] + 2 = 6 =dp[8] = 6
- 도시 B:
i - 5 = 3
→ dp[3] + 4 = 6 =dp[8] = 6
- 도시 A:
- i=9:
- 도시 A:
i - 3 - 3 - 3 = 0
→dp[9] = 6
- 도시 B:
i - 5 = 4
→ dp[4] + 4 = INF + 4
- 도시 A:
- i=10:
- 도시 A:
i - 3 = 7
→ dp[7] + 2 = INF + 2 - 도시 B:
i - 5 = 5
→ dp[5] + 4 =dp[10] = 8
- 도시 A:
- 따라서, 고객을 모을 수 있는 최소 비용은 8이 된다.
- 경우1: dp[10] = 8 → 10명, 비용 8
- 경우2: dp[3] + dp[5] + dp[3] = 8 → 11명, 비용 8
▪︎ 코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import java.io.*;
import java.util.*;
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int C = Integer.parseInt(st.nextToken()); // 목표 고객 수
int N = Integer.parseInt(st.nextToken()); // 도시 개수
int[] cost = new int[N]; // 홍보 비용
int[] customer = new int[N]; // 홍보 후 고객 증가 수
for (int i = 0; i < N; i++) {
st = new StringTokenizer(br.readLine());
cost[i] = Integer.parseInt(st.nextToken());
customer[i] = Integer.parseInt(st.nextToken());
}
// dp[i] = i명의 고객을 확보하기 위한 최소 비용
int INF = 100000; // 초기값은 매우 큰 값(INF)으로 설정 (최소값을 찾기 위해)
int[] dp = new int[C + 101]; // C 이상을 고려하기 위해 C+100까지 공간 확보
Arrays.fill(dp, INF); // dp 배열 INF로 초기화
dp[0] = 0; // 0명의 고객을 확보하는 데 필요한 비용은 0원
for (int i = 1; i < dp.length; i++) { // 확보 고객 수 i
for (int j = 0; j < N; j++) { // 홍보할 도시 j
if (i - customer[j] >= 0) { // 현재 고객 수 i에서 홍보 후 고객 수를 뺀 값이 0 이상일 때만 갱신 가능
// 기존 dp[i] vs (i - customer[j])명의 고객을 확보 후 도시 j를 한 번 더 사용하는 경우
dp[i] = Math.min(dp[i], dp[i - customer[j]] + cost[j]);
}
}
}
// C명 이상 확보한 경우 중 최소 비용 찾기
int result = INF;
for (int i = C; i < dp.length; i++) { // C 이상 고객 확보 가능한 모든 경우 중
result = Math.min(result, dp[i]); // 최소 비용을 선택
}
System.out.println(result);
}
}
- 도시 j를 마지막으로 사용한다고 가정하면, (모든 경우를 고려하기 위해 반복문 돌면서 가정)
- 그 전에
(i - customer[j])
명 확보하는 데 비용이dp[i - customer[j]]
(즉, 도시 j를 사용하기 전에 이미 확보한 고객 수 상태) - 거기에 도시 j를 한 번 더 쓰면 비용은
+ cost[j]
- 그 전에
- 기존 값(
dp[i]
)과 비교해서 더 작은 값을 저장한다
▪︎ 시간복잡도
O(N × C)
▪︎ 틀린 이유
- HashMap으로 도시 정보를 저장함 → 도시를 한 번만 사용 가능하다고 가정
- 문제에서는 같은 도시를 여러 번 홍보 가능(무한 배낭 문제)
HashMap
은 같은 key(고객 수)를 덮어쓰기 때문에 중복 홍보 고려 불가
- DP(동적 계획법) 로직을 생각하지 못함
- 문제는 최소 비용을 구해야 하는데
- 단순 반복이나 while문으로는 모든 경우를 탐색하기 어렵다.
dp[i]
배열을 사용해 최소 비용을 점진적으로 갱신해야 한다.
- C명 이상 확보하는 경우를 고려하지 않음
- 목표는 적어도 C명이므로 정확히 C명이 아니라, C보다 큰 고객 수도 계산해야 한다.
- 도시 고객 수가 i보다 클 때 처리 로직 부재
i - customer[j]
가 음수가 되면 이전 상태가 없으므로 갱신 불가- 이 경우는 DP에서 안전하게 제외해야 한다.
▪︎ 느낀점 / 기억할 정보
- dp란?
- 큰 문제를 작은 문제로 나누어 해결하고, 그 결과를 재사용하는 방법
- 배열(
dp[]
)에 부분 문제의 결과를 저장해 두고, 나중에 다시 계산하지 않고 꺼내 씀
dp[i]
= 고객 i명을 확보하기 위한 최소 비용무한 배낭 문제
1 2 3 4 5 6 7
for (int i = 1; i < dp.length; i++) { for (int j = 0; j < N; j++) { // 도시 j는 무제한 사용 가능 if (i - customer[j] >= 0) { dp[i] = Math.min(dp[i], dp[i - customer[j]] + cost[j]); } } }
This post is licensed under CC BY 4.0 by the author.