[백준 알고리즘]/[C++]
백준 1516 게임 개발 | C++
말하는펭귄
2021. 3. 26. 12:51
728x90
반응형
이번 포스팅은 백준 1516번 게임 개발입니다.
아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.
1516번: 게임 개발
첫째 줄에 건물의 종류 수 N(1 ≤ N ≤ 500)이 주어진다. 다음 N개의 줄에는 각 건물을 짓는데 걸리는 시간과 그 건물을 짓기 위해 먼저 지어져야 하는 건물들의 번호가 주어진다. 건물의 번호는 1부
www.acmicpc.net
기본 알고리즘
위상 정렬 Topological Sort
풀이
최소 시간을 갱신하는 조건
line56
result[next] = max(result[next], result[cur] + time[next]);
정점 next의 최소 건설 시간은 선행 정점이 모두 지어진 시간에 자신의 건물을 짓는 시간을 더한 값이다.
정점4의 갱신 상황에서
정점 4는 정점1과 정점3이 모두 건설되어야 건설할 수 있다.
cur=3일 때
result[next] : 14 //cur=1일 때 갱신된 값 (10+4)
result[cur] + time[next] : 18 //14(=선행 정점3의 최소 시간) + 4(=건설 소요 시간)
최종 result[4] = 18
전체 코드
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
//백준1516 게임 개발
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int main() {
/*입력*/
int N;
cin >> N;
vector<int> v[501];
for (int i = 1; i <= N; i++) {
int input;
cin >> input;
while (input != -1) {
v[i].push_back(input);
cin >> input;
}
}
/*각 건물을 짓는데 걸리는 시간*/
int time[501];
for (int i = 1; i <= N; i++) {
time[i] = v[i][0];
}
/*위상 정렬*/
vector<int> adj[501];
int inDeg[501] = { 0, };
queue<int> q;
int result[501]; //출력배열 - 각 건물이 완성되기까지 걸리는 최소 시간
for (int i = 1; i <= N; i++) {
for (int j = 1; j < v[i].size(); j++) {
adj[v[i][j]].push_back(i);
inDeg[i]++;
}
}
for (int i = 1; i <= N; i++) {
if (inDeg[i] == 0) {
q.push(i);
}
result[i] = time[i];
}
while (!q.empty()) {
int cur = q.front();
q.pop();
for (int i = 0; i < adj[cur].size(); i++) {
int next = adj[cur][i];
inDeg[next]--;
result[next] = max(result[next], result[cur] + time[next]); //★
if (inDeg[next] == 0) {
q.push(next);
}
}
}
/*출력*/
for (int i = 1; i <= N; i++) {
cout << result[i] << endl;
}
return 0;
}
|
cs |
728x90
반응형