반응형

 

https://www.acmicpc.net/problem/1005

 

1005번: ACM Craft

첫째 줄에는 테스트케이스의 개수 T가 주어진다. 각 테스트 케이스는 다음과 같이 주어진다. 첫째 줄에 건물의 개수 N 과 건물간의 건설순서규칙의 총 개수 K이 주어진다. (건물의 번호는 1번부

www.acmicpc.net

0123456

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
71
72
#include <iostream>    
#include <vector>
#include <queue>
#include <algorithm>
#include <cstring>
#define rep(i,n) for(int i=1;i<=n;i++)
using namespace std;
vector<vector<int> > v;
queue<int> q;
int t, n, k, target;
int build_time[1001];
int in_degree[1001];
int dp[1001];
int before[1001];
void func();
void init();
void input();
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> t;
    while (t--) {
        input();
        func();
        cout << dp[target] << '\n';
    }
}
 
void func() {
    rep(i, n) {
        if (in_degree[i] == 0
            q.push(i);
        dp[i] = build_time[i];
    }
 
    while (!q.empty()) {
        int now = q.front(); q.pop();
        for (int next : v[now]) {
            in_degree[next]--;
            before[next] = max(before[next], dp[now]);
            if (in_degree[next] == 0) {
                q.push(next);
                dp[next] += before[next];
                if (next == target) return;
            }
        }
    }
}
 
void init() {
    v.clear();
    v.resize(n + 1);
    while (!q.empty()) q.pop();
    memset(build_time, 0sizeof(build_time));
    memset(in_degree, 0sizeof(in_degree));
    memset(dp, 0sizeof(dp));
    memset(before, 0sizeof(before));
}
 
void input() {
    cin >> n >> k;
    init();
    rep(i, n)
        cin >> build_time[i];
    rep(i, k) {
        int a, b;
        cin >> a >> b;
        v[a].push_back(b);
        in_degree[b]++;
    }
    cin >> target;
}
cs
반응형

'백준 > 트리DP' 카테고리의 다른 글

백준 2213  (0) 2021.08.02
백준 2533 [복습필수]  (0) 2021.08.01
백준 15681  (0) 2021.07.30

+ Recent posts