#include <bits/stdc++.h>
#include <cassert>
using namespace std;
#define all(v) v.begin(), v.end()
typedef long long ll;
const int NMAX = 1e5 + 5;
set<pair<ll, ll>> adj[NMAX];
int dg[NMAX], vis[NMAX];
ll dp[NMAX][2], S[NMAX];
vector<int> D[NMAX];
set<int> X;
multiset<ll> A[NMAX], t[NMAX];
void dfs(int now, int k) {
vis[now] = 1;
ll& ret = dp[now][0] = 0;
ll c = 0;
multiset<ll> C;
for(auto& p : adj[now]) {
ll nx = p.first, w = p.second;
if(vis[nx]) continue;
dfs(nx, k);
ret += dp[nx][0] + w;
ll t = dp[nx][1] - dp[nx][0] - w;
c++;
C.emplace(t);
}
while((int)A[now].size() < k - c) {
auto it = t[now].begin();
assert(*it <= 0);
A[now].emplace(*it); S[now] += *it;
t[now].erase(it);
}
while(A[now].size() && (int)A[now].size() > k - c) {
auto it = --A[now].end(); S[now] -= *it;
assert(*it <= 0);
t[now].emplace(*it);
A[now].erase(it);
}
if(c >= k) assert(A[now].size() == 0);
else assert(A[now].size() + c == k);
ret += S[now];
/*cout << "now is : " << now << ' ' << ret << '\n';
for (ll x : A[now]) cout << x << ' ';
cout << '\n';
for (ll x : C[now]) cout << x << ' ';
cout << '\n';*/
auto it = t[now].begin();
for(int i = 0; i < c && it != t[now].end(); i++) C.emplace(*it++);
if(!c) {
//cout << k << ' ' << now << ' ' << ret << '\n';
//cout << ret << '\n';
dp[now][1] = ret - *A[now].rbegin();
return;
}
for(ll x : A[now]) {
C.emplace(x); ret -= x;
}
it = C.begin();
int sz = k - 1;
while(sz--) ret += min(*it++, 0LL);
dp[now][1] = ret;
ret += min(*it, 0LL);
//cout << ret << '\n';
//cout << "Now is : " << now << '\n';
//for(ll x : C[now]) cout << x << ' ';
//cout << '\n';
return;
}
vector<ll> minimum_closure_costs(int N, vector<int> U, vector<int> V, vector<int> W) {
vector<ll> ans(N, 0);
for(int i = 0; i < N - 1; i++) {
adj[U[i]].emplace(V[i], W[i]);
adj[V[i]].emplace(U[i], W[i]);
dg[U[i]]++; dg[V[i]]++;
ans[0] += W[i];
}
for(int i = 0; i < N; i++) {
D[dg[i]].emplace_back(i);
X.emplace(i);
}
for(int k = 1; k < N; k++) {
// 노드 삭제하기
//cout << "NOW TURN IS : " << k << '\n';
for(int x : D[k]) {
X.erase(x);
//cout << "erase : " << x << '\n';
for(auto& p : adj[x]) {
ll nx = p.first, w = p.second;
adj[nx].erase({ x, w });
S[nx] += w; t[nx].emplace(-w);
}
}
// 각 트리에 대해서 탐색
for(int x : X) if(!vis[x]) {
dfs(x, k);
ans[k] += dp[x][0];
}
for(int x : X) vis[x] = 0;
}
return ans;
}
/*
5
0 1 1
0 2 4
0 3 3
2 4 2
*/
/*
4
0 1 5
0 3 5
0 2 10
*/
/*
5
0 1 5
1 2 6
2 3 1
3 4 8
*/
/*
5
0 1 1
0 2 2
0 3 3
0 4 4
*/
//int main(void) {
// ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
//
// freopen("input.txt", "r", stdin);
// int N;
// cin >> N;
// vector<int> U(N), V(N), W(N);
// for(int i = 0; i < N - 1; i++) cin >> U[i] >> V[i] >> W[i];
// vector<ll> ans = minimum_closure_costs(N, U, V, W);
// for(int i = 0; i < N; i++) cout << ans[i] << ' ';
// return 0;
//}
Compilation message
In file included from /usr/include/c++/10/cassert:44,
from roads.cpp:2:
roads.cpp: In function 'void dfs(int, int)':
roads.cpp:45:32: warning: comparison of integer expressions of different signedness: 'long long unsigned int' and 'int' [-Wsign-compare]
45 | else assert(A[now].size() + c == k);
| ~~~~~~~~~~~~~~~~~~^~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
16716 KB |
Output is correct |
2 |
Correct |
11 ms |
17100 KB |
Output is correct |
3 |
Correct |
11 ms |
17120 KB |
Output is correct |
4 |
Correct |
11 ms |
17204 KB |
Output is correct |
5 |
Correct |
9 ms |
16716 KB |
Output is correct |
6 |
Correct |
9 ms |
16716 KB |
Output is correct |
7 |
Correct |
9 ms |
16716 KB |
Output is correct |
8 |
Correct |
13 ms |
17100 KB |
Output is correct |
9 |
Correct |
11 ms |
17100 KB |
Output is correct |
10 |
Correct |
8 ms |
16716 KB |
Output is correct |
11 |
Correct |
9 ms |
16732 KB |
Output is correct |
12 |
Correct |
107 ms |
29416 KB |
Output is correct |
13 |
Correct |
189 ms |
37896 KB |
Output is correct |
14 |
Correct |
175 ms |
35752 KB |
Output is correct |
15 |
Correct |
196 ms |
37952 KB |
Output is correct |
16 |
Correct |
208 ms |
38328 KB |
Output is correct |
17 |
Correct |
217 ms |
38452 KB |
Output is correct |
18 |
Correct |
8 ms |
16716 KB |
Output is correct |
19 |
Correct |
174 ms |
35848 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
16716 KB |
Output is correct |
2 |
Correct |
95 ms |
52764 KB |
Output is correct |
3 |
Correct |
105 ms |
56560 KB |
Output is correct |
4 |
Correct |
112 ms |
59608 KB |
Output is correct |
5 |
Correct |
110 ms |
59708 KB |
Output is correct |
6 |
Correct |
10 ms |
17484 KB |
Output is correct |
7 |
Correct |
10 ms |
17588 KB |
Output is correct |
8 |
Correct |
10 ms |
17484 KB |
Output is correct |
9 |
Correct |
9 ms |
16716 KB |
Output is correct |
10 |
Correct |
8 ms |
16716 KB |
Output is correct |
11 |
Correct |
9 ms |
16716 KB |
Output is correct |
12 |
Correct |
66 ms |
42260 KB |
Output is correct |
13 |
Correct |
108 ms |
59352 KB |
Output is correct |
14 |
Correct |
8 ms |
16716 KB |
Output is correct |
15 |
Correct |
96 ms |
55152 KB |
Output is correct |
16 |
Correct |
108 ms |
59644 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
16716 KB |
Output is correct |
2 |
Correct |
8 ms |
16716 KB |
Output is correct |
3 |
Correct |
8 ms |
16716 KB |
Output is correct |
4 |
Correct |
8 ms |
16716 KB |
Output is correct |
5 |
Correct |
10 ms |
16812 KB |
Output is correct |
6 |
Correct |
9 ms |
16716 KB |
Output is correct |
7 |
Correct |
9 ms |
16772 KB |
Output is correct |
8 |
Correct |
9 ms |
16736 KB |
Output is correct |
9 |
Correct |
9 ms |
16716 KB |
Output is correct |
10 |
Correct |
9 ms |
16740 KB |
Output is correct |
11 |
Correct |
10 ms |
16832 KB |
Output is correct |
12 |
Correct |
8 ms |
16740 KB |
Output is correct |
13 |
Correct |
9 ms |
16716 KB |
Output is correct |
14 |
Correct |
9 ms |
16716 KB |
Output is correct |
15 |
Correct |
8 ms |
16736 KB |
Output is correct |
16 |
Correct |
9 ms |
16716 KB |
Output is correct |
17 |
Correct |
8 ms |
16716 KB |
Output is correct |
18 |
Correct |
8 ms |
16740 KB |
Output is correct |
19 |
Correct |
8 ms |
16716 KB |
Output is correct |
20 |
Correct |
8 ms |
16716 KB |
Output is correct |
21 |
Correct |
8 ms |
16716 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
16716 KB |
Output is correct |
2 |
Correct |
8 ms |
16716 KB |
Output is correct |
3 |
Correct |
8 ms |
16716 KB |
Output is correct |
4 |
Correct |
8 ms |
16716 KB |
Output is correct |
5 |
Correct |
10 ms |
16812 KB |
Output is correct |
6 |
Correct |
9 ms |
16716 KB |
Output is correct |
7 |
Correct |
9 ms |
16772 KB |
Output is correct |
8 |
Correct |
9 ms |
16736 KB |
Output is correct |
9 |
Correct |
9 ms |
16716 KB |
Output is correct |
10 |
Correct |
9 ms |
16740 KB |
Output is correct |
11 |
Correct |
10 ms |
16832 KB |
Output is correct |
12 |
Correct |
8 ms |
16740 KB |
Output is correct |
13 |
Correct |
9 ms |
16716 KB |
Output is correct |
14 |
Correct |
9 ms |
16716 KB |
Output is correct |
15 |
Correct |
8 ms |
16736 KB |
Output is correct |
16 |
Correct |
9 ms |
16716 KB |
Output is correct |
17 |
Correct |
8 ms |
16716 KB |
Output is correct |
18 |
Correct |
8 ms |
16740 KB |
Output is correct |
19 |
Correct |
8 ms |
16716 KB |
Output is correct |
20 |
Correct |
8 ms |
16716 KB |
Output is correct |
21 |
Correct |
8 ms |
16716 KB |
Output is correct |
22 |
Correct |
9 ms |
16716 KB |
Output is correct |
23 |
Correct |
10 ms |
16972 KB |
Output is correct |
24 |
Correct |
13 ms |
17252 KB |
Output is correct |
25 |
Correct |
11 ms |
17100 KB |
Output is correct |
26 |
Correct |
13 ms |
17264 KB |
Output is correct |
27 |
Correct |
15 ms |
17228 KB |
Output is correct |
28 |
Correct |
20 ms |
17136 KB |
Output is correct |
29 |
Correct |
10 ms |
17228 KB |
Output is correct |
30 |
Correct |
14 ms |
17228 KB |
Output is correct |
31 |
Correct |
12 ms |
17228 KB |
Output is correct |
32 |
Correct |
24 ms |
17260 KB |
Output is correct |
33 |
Correct |
10 ms |
17484 KB |
Output is correct |
34 |
Correct |
10 ms |
17544 KB |
Output is correct |
35 |
Correct |
9 ms |
17492 KB |
Output is correct |
36 |
Correct |
10 ms |
17100 KB |
Output is correct |
37 |
Correct |
11 ms |
17228 KB |
Output is correct |
38 |
Correct |
11 ms |
17160 KB |
Output is correct |
39 |
Correct |
8 ms |
16760 KB |
Output is correct |
40 |
Correct |
8 ms |
16688 KB |
Output is correct |
41 |
Correct |
8 ms |
16728 KB |
Output is correct |
42 |
Correct |
11 ms |
16844 KB |
Output is correct |
43 |
Correct |
9 ms |
16736 KB |
Output is correct |
44 |
Correct |
9 ms |
16740 KB |
Output is correct |
45 |
Correct |
9 ms |
16740 KB |
Output is correct |
46 |
Correct |
9 ms |
16756 KB |
Output is correct |
47 |
Correct |
9 ms |
16716 KB |
Output is correct |
48 |
Correct |
9 ms |
16716 KB |
Output is correct |
49 |
Correct |
9 ms |
16804 KB |
Output is correct |
50 |
Correct |
8 ms |
16716 KB |
Output is correct |
51 |
Correct |
10 ms |
16736 KB |
Output is correct |
52 |
Correct |
8 ms |
16744 KB |
Output is correct |
53 |
Correct |
10 ms |
17172 KB |
Output is correct |
54 |
Correct |
11 ms |
17228 KB |
Output is correct |
55 |
Correct |
20 ms |
17200 KB |
Output is correct |
56 |
Correct |
9 ms |
17056 KB |
Output is correct |
57 |
Correct |
10 ms |
17120 KB |
Output is correct |
58 |
Correct |
8 ms |
16712 KB |
Output is correct |
59 |
Correct |
9 ms |
16808 KB |
Output is correct |
60 |
Correct |
8 ms |
16716 KB |
Output is correct |
61 |
Correct |
8 ms |
16716 KB |
Output is correct |
62 |
Correct |
8 ms |
16716 KB |
Output is correct |
63 |
Correct |
8 ms |
16716 KB |
Output is correct |
64 |
Correct |
10 ms |
17264 KB |
Output is correct |
65 |
Correct |
10 ms |
17256 KB |
Output is correct |
66 |
Correct |
12 ms |
17124 KB |
Output is correct |
67 |
Correct |
25 ms |
17228 KB |
Output is correct |
68 |
Correct |
13 ms |
17264 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
232 ms |
40404 KB |
Output is correct |
2 |
Correct |
210 ms |
39532 KB |
Output is correct |
3 |
Correct |
1010 ms |
39740 KB |
Output is correct |
4 |
Correct |
223 ms |
40764 KB |
Output is correct |
5 |
Correct |
268 ms |
39868 KB |
Output is correct |
6 |
Correct |
1070 ms |
38436 KB |
Output is correct |
7 |
Correct |
272 ms |
40768 KB |
Output is correct |
8 |
Correct |
179 ms |
36252 KB |
Output is correct |
9 |
Correct |
218 ms |
46444 KB |
Output is correct |
10 |
Correct |
233 ms |
40424 KB |
Output is correct |
11 |
Correct |
795 ms |
40512 KB |
Output is correct |
12 |
Execution timed out |
2051 ms |
38756 KB |
Time limit exceeded |
13 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
232 ms |
40404 KB |
Output is correct |
2 |
Correct |
210 ms |
39532 KB |
Output is correct |
3 |
Correct |
1010 ms |
39740 KB |
Output is correct |
4 |
Correct |
223 ms |
40764 KB |
Output is correct |
5 |
Correct |
268 ms |
39868 KB |
Output is correct |
6 |
Correct |
1070 ms |
38436 KB |
Output is correct |
7 |
Correct |
272 ms |
40768 KB |
Output is correct |
8 |
Correct |
179 ms |
36252 KB |
Output is correct |
9 |
Correct |
218 ms |
46444 KB |
Output is correct |
10 |
Correct |
233 ms |
40424 KB |
Output is correct |
11 |
Correct |
795 ms |
40512 KB |
Output is correct |
12 |
Execution timed out |
2051 ms |
38756 KB |
Time limit exceeded |
13 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
16716 KB |
Output is correct |
2 |
Correct |
11 ms |
17100 KB |
Output is correct |
3 |
Correct |
11 ms |
17120 KB |
Output is correct |
4 |
Correct |
11 ms |
17204 KB |
Output is correct |
5 |
Correct |
9 ms |
16716 KB |
Output is correct |
6 |
Correct |
9 ms |
16716 KB |
Output is correct |
7 |
Correct |
9 ms |
16716 KB |
Output is correct |
8 |
Correct |
13 ms |
17100 KB |
Output is correct |
9 |
Correct |
11 ms |
17100 KB |
Output is correct |
10 |
Correct |
8 ms |
16716 KB |
Output is correct |
11 |
Correct |
9 ms |
16732 KB |
Output is correct |
12 |
Correct |
107 ms |
29416 KB |
Output is correct |
13 |
Correct |
189 ms |
37896 KB |
Output is correct |
14 |
Correct |
175 ms |
35752 KB |
Output is correct |
15 |
Correct |
196 ms |
37952 KB |
Output is correct |
16 |
Correct |
208 ms |
38328 KB |
Output is correct |
17 |
Correct |
217 ms |
38452 KB |
Output is correct |
18 |
Correct |
8 ms |
16716 KB |
Output is correct |
19 |
Correct |
174 ms |
35848 KB |
Output is correct |
20 |
Correct |
8 ms |
16716 KB |
Output is correct |
21 |
Correct |
95 ms |
52764 KB |
Output is correct |
22 |
Correct |
105 ms |
56560 KB |
Output is correct |
23 |
Correct |
112 ms |
59608 KB |
Output is correct |
24 |
Correct |
110 ms |
59708 KB |
Output is correct |
25 |
Correct |
10 ms |
17484 KB |
Output is correct |
26 |
Correct |
10 ms |
17588 KB |
Output is correct |
27 |
Correct |
10 ms |
17484 KB |
Output is correct |
28 |
Correct |
9 ms |
16716 KB |
Output is correct |
29 |
Correct |
8 ms |
16716 KB |
Output is correct |
30 |
Correct |
9 ms |
16716 KB |
Output is correct |
31 |
Correct |
66 ms |
42260 KB |
Output is correct |
32 |
Correct |
108 ms |
59352 KB |
Output is correct |
33 |
Correct |
8 ms |
16716 KB |
Output is correct |
34 |
Correct |
96 ms |
55152 KB |
Output is correct |
35 |
Correct |
108 ms |
59644 KB |
Output is correct |
36 |
Correct |
8 ms |
16716 KB |
Output is correct |
37 |
Correct |
8 ms |
16716 KB |
Output is correct |
38 |
Correct |
8 ms |
16716 KB |
Output is correct |
39 |
Correct |
8 ms |
16716 KB |
Output is correct |
40 |
Correct |
10 ms |
16812 KB |
Output is correct |
41 |
Correct |
9 ms |
16716 KB |
Output is correct |
42 |
Correct |
9 ms |
16772 KB |
Output is correct |
43 |
Correct |
9 ms |
16736 KB |
Output is correct |
44 |
Correct |
9 ms |
16716 KB |
Output is correct |
45 |
Correct |
9 ms |
16740 KB |
Output is correct |
46 |
Correct |
10 ms |
16832 KB |
Output is correct |
47 |
Correct |
8 ms |
16740 KB |
Output is correct |
48 |
Correct |
9 ms |
16716 KB |
Output is correct |
49 |
Correct |
9 ms |
16716 KB |
Output is correct |
50 |
Correct |
8 ms |
16736 KB |
Output is correct |
51 |
Correct |
9 ms |
16716 KB |
Output is correct |
52 |
Correct |
8 ms |
16716 KB |
Output is correct |
53 |
Correct |
8 ms |
16740 KB |
Output is correct |
54 |
Correct |
8 ms |
16716 KB |
Output is correct |
55 |
Correct |
8 ms |
16716 KB |
Output is correct |
56 |
Correct |
8 ms |
16716 KB |
Output is correct |
57 |
Correct |
9 ms |
16716 KB |
Output is correct |
58 |
Correct |
10 ms |
16972 KB |
Output is correct |
59 |
Correct |
13 ms |
17252 KB |
Output is correct |
60 |
Correct |
11 ms |
17100 KB |
Output is correct |
61 |
Correct |
13 ms |
17264 KB |
Output is correct |
62 |
Correct |
15 ms |
17228 KB |
Output is correct |
63 |
Correct |
20 ms |
17136 KB |
Output is correct |
64 |
Correct |
10 ms |
17228 KB |
Output is correct |
65 |
Correct |
14 ms |
17228 KB |
Output is correct |
66 |
Correct |
12 ms |
17228 KB |
Output is correct |
67 |
Correct |
24 ms |
17260 KB |
Output is correct |
68 |
Correct |
10 ms |
17484 KB |
Output is correct |
69 |
Correct |
10 ms |
17544 KB |
Output is correct |
70 |
Correct |
9 ms |
17492 KB |
Output is correct |
71 |
Correct |
10 ms |
17100 KB |
Output is correct |
72 |
Correct |
11 ms |
17228 KB |
Output is correct |
73 |
Correct |
11 ms |
17160 KB |
Output is correct |
74 |
Correct |
8 ms |
16760 KB |
Output is correct |
75 |
Correct |
8 ms |
16688 KB |
Output is correct |
76 |
Correct |
8 ms |
16728 KB |
Output is correct |
77 |
Correct |
11 ms |
16844 KB |
Output is correct |
78 |
Correct |
9 ms |
16736 KB |
Output is correct |
79 |
Correct |
9 ms |
16740 KB |
Output is correct |
80 |
Correct |
9 ms |
16740 KB |
Output is correct |
81 |
Correct |
9 ms |
16756 KB |
Output is correct |
82 |
Correct |
9 ms |
16716 KB |
Output is correct |
83 |
Correct |
9 ms |
16716 KB |
Output is correct |
84 |
Correct |
9 ms |
16804 KB |
Output is correct |
85 |
Correct |
8 ms |
16716 KB |
Output is correct |
86 |
Correct |
10 ms |
16736 KB |
Output is correct |
87 |
Correct |
8 ms |
16744 KB |
Output is correct |
88 |
Correct |
10 ms |
17172 KB |
Output is correct |
89 |
Correct |
11 ms |
17228 KB |
Output is correct |
90 |
Correct |
20 ms |
17200 KB |
Output is correct |
91 |
Correct |
9 ms |
17056 KB |
Output is correct |
92 |
Correct |
10 ms |
17120 KB |
Output is correct |
93 |
Correct |
8 ms |
16712 KB |
Output is correct |
94 |
Correct |
9 ms |
16808 KB |
Output is correct |
95 |
Correct |
8 ms |
16716 KB |
Output is correct |
96 |
Correct |
8 ms |
16716 KB |
Output is correct |
97 |
Correct |
8 ms |
16716 KB |
Output is correct |
98 |
Correct |
8 ms |
16716 KB |
Output is correct |
99 |
Correct |
10 ms |
17264 KB |
Output is correct |
100 |
Correct |
10 ms |
17256 KB |
Output is correct |
101 |
Correct |
12 ms |
17124 KB |
Output is correct |
102 |
Correct |
25 ms |
17228 KB |
Output is correct |
103 |
Correct |
13 ms |
17264 KB |
Output is correct |
104 |
Correct |
232 ms |
40404 KB |
Output is correct |
105 |
Correct |
210 ms |
39532 KB |
Output is correct |
106 |
Correct |
1010 ms |
39740 KB |
Output is correct |
107 |
Correct |
223 ms |
40764 KB |
Output is correct |
108 |
Correct |
268 ms |
39868 KB |
Output is correct |
109 |
Correct |
1070 ms |
38436 KB |
Output is correct |
110 |
Correct |
272 ms |
40768 KB |
Output is correct |
111 |
Correct |
179 ms |
36252 KB |
Output is correct |
112 |
Correct |
218 ms |
46444 KB |
Output is correct |
113 |
Correct |
233 ms |
40424 KB |
Output is correct |
114 |
Correct |
795 ms |
40512 KB |
Output is correct |
115 |
Execution timed out |
2051 ms |
38756 KB |
Time limit exceeded |
116 |
Halted |
0 ms |
0 KB |
- |