반응형

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

 

2213번: 트리의 독립집합

첫째 줄에 트리의 정점의 수 n이 주어진다. n은 10,000이하인 양의 정수이다. 1부터 n사이의 정수가 트리의 정점이라고 가정한다. 둘째 줄에는 n개의 정수 w1, w2, ..., wn이 주어지는데, wi는 정점 i의

www.acmicpc.net

1. 최댓값 찾기

012345

2. 출력

01234

소스코드 

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
73
74
75
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#define rep(i,n) for(int i=1;i<=n;i++)
#define pii pair<int, int>
 
using namespace std;
vector<vector<int> > v;
vector<pii> child[10001][2];
priority_queue<int> pq;
int n, dp[10001][2], a,b;
bool visited[10001];
void input();
void print();
void dfs(int now) {
    for (int next : v[now]) {
        if (!visited[next]) {
            visited[next] = 1;
            dfs(next);
 
            // now 미포함
            if (dp[next][0] > dp[next][1]) {
                dp[now][0] += dp[next][0];
                child[now][0].push_back({ next,0 });
            }
            else {
                dp[now][0] += dp[next][1];
                child[now][0].push_back({ next,1 });
            }
 
            // now 포함
            dp[now][1] += dp[next][0];
            child[now][1].push_back({ next, 0 });
        }
    }
}
 
void trace(int now, int idx) {
    if (idx == 1)
        pq.push(-now);
    for (pii next : child[now][idx]) 
        trace(next.first, next.second);
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    input();
    visited[1] = 1;
    dfs(1);
 
    if (dp[1][0] > dp[1][1]) trace(1, 0);
    else trace(1, 1);
 
    print();
}
 
void input() {
    cin >> n;
    v.resize(n + 1);
    rep(i, n)
        cin >> dp[i][1];
    rep(i, n - 1) {
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
}
void print() {
    cout << max(dp[1][0], dp[1][1]) << '\n';
    while (!pq.empty()) {
        cout << -pq.top() << ' ';
        pq.pop();
    }
}
 cs
반응형

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

백준 2533 [복습필수]  (0) 2021.08.01
백준 15681  (0) 2021.07.30
백준 1005  (0) 2021.07.24
반응형

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

 

2533번: 사회망 서비스(SNS)

페이스북, 트위터, 카카오톡과 같은 사회망 서비스(SNS)가 널리 사용됨에 따라, 사회망을 통하여 사람들이 어떻게 새로운 아이디어를 받아들이게 되는가를 이해하는 문제가 중요해졌다. 사회망

www.acmicpc.net

DP

dp[i][0]: 본인이 얼리어답터가 아님 -> 이웃 모두 얼리어답터
dp[i][1]: 본인이 얼리어답터임        -> 이웃 얼리어답터 상관 없이 작은 값 

 

점화식

dp[now][0] += dp[next][1]

dp[now][1] += min(dp[next][0], dp[next][1])

 

간선 양방향 및 visited 쓰는 이유

간선 단방향으로 입력할 경우, 간선 a, b 순서가 바뀌면 값이 달라짐

8

1 2

1 3

1 4

2 5

2 6

4 7

4 8

 

8
2 1
3 1
4 1
5 2
6 2
7 4
8 4

진행과정

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
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#define rep(i,n) for(int i=1;i<=n;i++)
using namespace std;
vector<vector<int> > v;
int n, a, b, visited[1000001];
int dp[1000001][2];    // dp[i][0]: 본인이 얼리어답터가 아님    -> 이웃 모두 얼리어답터
                    // dp[i][1]: 본인이 얼리어답터임        -> 얼리어답터 상관없음
void func(int now) {
    dp[now][1] = 1;
    for (int next : v[now]) {
        if (!visited[next]) {
            visited[next] = 1;
            func(next);
            dp[now][0] += dp[next][1];
            dp[now][1] += min(dp[next][0], dp[next][1]);
        }
    }
}
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n;
    v.resize(n + 1);
    rep(i, n-1) {
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    visited[1] = 1;
    func(1);
    cout << min(dp[1][0], dp[1][1]);
}
 cs
반응형

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

백준 2213  (0) 2021.08.02
백준 15681  (0) 2021.07.30
백준 1005  (0) 2021.07.24
반응형

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

 

15681번: 트리와 쿼리

트리의 정점의 수 N과 루트의 번호 R, 쿼리의 수 Q가 주어진다. (2 ≤ N ≤ 105, 1 ≤ R ≤ N, 1 ≤ Q ≤ 105) 이어 N-1줄에 걸쳐, U V의 형태로 트리에 속한 간선의 정보가 주어진다. (1 ≤ U, V ≤ N, U ≠ V)

www.acmicpc.net

DFS로 간단하게 해결 가능

 

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
#include <iostream>
#include <vector>
#define rep(i,n) for(int i=1;i<=n;i++)
using namespace std;
vector<vector<int> > v;
int n, r, q, a, b,x, child[100001], visited[100001];
void input();
void print();
int func(int r) {
    for (int next : v[r]) {
        if (!visited[next]) {
            visited[next] = 1;
            child[r]+=func(next);
        }
    }
    return child[r] + 1;
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    input();
    func(r);
    print();
}
void input() {
    cin >> n >> r >> q;
    v.resize(n + 1);
    rep(i, n - 1) {
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    visited[r] = 1;
}
void print() {
    rep(i, q) {
        cin >> x;
        cout << child[x] + 1 << '\n';
    }
}
 cs
반응형

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

백준 2213  (0) 2021.08.02
백준 2533 [복습필수]  (0) 2021.08.01
백준 1005  (0) 2021.07.24
반응형

 

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, 0, sizeof(build_time));
    memset(in_degree, 0, sizeof(in_degree));
    memset(dp, 0, sizeof(dp));
    memset(before, 0, sizeof(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

티스토리툴바