#include "roads.h"
#include <bits/stdc++.h>
#define fi first
#define se second
#define nmax 100008
const long long inf = 1e18;
using namespace std;
typedef pair<long long, long long> pii;
long long n, sub1 = true, sub2 = true;
long long dp[2008][2008][2];
long long f[5][nmax][2];
vector<pii> l[nmax];
long long sdq(int id, int x, int pre, int kt){
if (f[id][x][kt] != -1)
return f[id][x][kt];
if (!id && kt)
return f[id][x][kt] = inf;
long long res = 0;
vector<long long>vt;
for (pii it : l[x])
if (it.fi != pre){
res += sdq(id, it.fi, x, 1);
vt.push_back(sdq(id, it.fi, x, 0) + it.se - sdq(id, it.fi, x, 1));
}
sort(vt.begin(), vt.end(), greater<long long>());
while (!vt.empty() && ((int)vt.size() + kt > id || vt.back() < 0)){
res += vt.back();
vt.pop_back();
}
return f[id][x][kt] = res;
}
long long dq(int id, int x, int pre, int kt){
if (dp[id][x][kt] != -1)
return dp[id][x][kt];
if (!id && kt)
return dp[id][x][kt] = inf;
long long res = 0;
vector<long long>vt;
for (pii it : l[x])
if (it.fi != pre){
res += dq(id, it.fi, x, 1);
vt.push_back(dq(id, it.fi, x, 0) + it.se - dq(id, it.fi, x, 1));
}
sort(vt.begin(), vt.end(), greater<long long>());
while (!vt.empty() && ((int)vt.size() + kt > id || vt.back() < 0)){
res += vt.back();
vt.pop_back();
}
return dp[id][x][kt] = res;
}
std::vector<long long> minimum_closure_costs(int N, std::vector<int> U, std::vector<int> V, std::vector<int> W) {
n = N;
for (long long i = 0; i < n - 1; i++){
if (U[i] != 0)
sub1 = false;
if (U[i] != i || V[i] != i + 1)
sub2 = false;
l[U[i]].push_back({V[i], W[i]});
l[V[i]].push_back({U[i], W[i]});
}
vector<long long>res;
if (sub1){
sort(W.begin(), W.end(), greater<int>());
long long j = 0;
for (int tmp : W)
j += tmp;
res.push_back(j);
for (long long i = 0; i < n - 1; i++){
j -= W[i];
res.push_back(j);
}
return res;
}
if (sub2){
memset(f, -1, sizeof(f));
for (long long k = 0; k < n; k++)
if (k < 2)
res.push_back(sdq(k, 0, 0, 0));
else
res.push_back(0);
return res;
}
assert(n <= 2000);
memset(dp, -1, sizeof(dp));
for (long long k = 0; k < n; k++)
res.push_back(dq(k, 0, 0, 0));
return res;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2772 KB |
Output is correct |
3 |
Correct |
2 ms |
2772 KB |
Output is correct |
4 |
Correct |
3 ms |
2772 KB |
Output is correct |
5 |
Correct |
2 ms |
2644 KB |
Output is correct |
6 |
Correct |
1 ms |
2644 KB |
Output is correct |
7 |
Correct |
2 ms |
2644 KB |
Output is correct |
8 |
Correct |
3 ms |
2772 KB |
Output is correct |
9 |
Correct |
2 ms |
2772 KB |
Output is correct |
10 |
Correct |
1 ms |
2644 KB |
Output is correct |
11 |
Correct |
2 ms |
2644 KB |
Output is correct |
12 |
Correct |
29 ms |
7848 KB |
Output is correct |
13 |
Correct |
51 ms |
11452 KB |
Output is correct |
14 |
Correct |
54 ms |
10904 KB |
Output is correct |
15 |
Correct |
58 ms |
11616 KB |
Output is correct |
16 |
Correct |
60 ms |
11812 KB |
Output is correct |
17 |
Correct |
58 ms |
11784 KB |
Output is correct |
18 |
Correct |
2 ms |
2644 KB |
Output is correct |
19 |
Correct |
48 ms |
10708 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
2644 KB |
Output is correct |
2 |
Correct |
69 ms |
31664 KB |
Output is correct |
3 |
Correct |
75 ms |
36256 KB |
Output is correct |
4 |
Correct |
77 ms |
37960 KB |
Output is correct |
5 |
Correct |
77 ms |
38088 KB |
Output is correct |
6 |
Correct |
6 ms |
10964 KB |
Output is correct |
7 |
Correct |
6 ms |
10996 KB |
Output is correct |
8 |
Correct |
6 ms |
10964 KB |
Output is correct |
9 |
Correct |
5 ms |
10580 KB |
Output is correct |
10 |
Correct |
5 ms |
10452 KB |
Output is correct |
11 |
Correct |
5 ms |
10452 KB |
Output is correct |
12 |
Correct |
44 ms |
26468 KB |
Output is correct |
13 |
Correct |
72 ms |
37244 KB |
Output is correct |
14 |
Correct |
1 ms |
2644 KB |
Output is correct |
15 |
Correct |
66 ms |
34676 KB |
Output is correct |
16 |
Correct |
71 ms |
37180 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
2644 KB |
Output is correct |
2 |
Correct |
26 ms |
65688 KB |
Output is correct |
3 |
Correct |
25 ms |
65756 KB |
Output is correct |
4 |
Correct |
28 ms |
65672 KB |
Output is correct |
5 |
Correct |
31 ms |
65748 KB |
Output is correct |
6 |
Correct |
30 ms |
65744 KB |
Output is correct |
7 |
Correct |
30 ms |
65680 KB |
Output is correct |
8 |
Correct |
29 ms |
65748 KB |
Output is correct |
9 |
Correct |
29 ms |
65748 KB |
Output is correct |
10 |
Correct |
5 ms |
10452 KB |
Output is correct |
11 |
Correct |
5 ms |
10452 KB |
Output is correct |
12 |
Correct |
5 ms |
10452 KB |
Output is correct |
13 |
Correct |
2 ms |
2644 KB |
Output is correct |
14 |
Correct |
2 ms |
2644 KB |
Output is correct |
15 |
Correct |
2 ms |
2632 KB |
Output is correct |
16 |
Correct |
28 ms |
65740 KB |
Output is correct |
17 |
Correct |
30 ms |
65748 KB |
Output is correct |
18 |
Correct |
29 ms |
65748 KB |
Output is correct |
19 |
Correct |
1 ms |
2644 KB |
Output is correct |
20 |
Correct |
2 ms |
2644 KB |
Output is correct |
21 |
Correct |
2 ms |
2644 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
2644 KB |
Output is correct |
2 |
Correct |
26 ms |
65688 KB |
Output is correct |
3 |
Correct |
25 ms |
65756 KB |
Output is correct |
4 |
Correct |
28 ms |
65672 KB |
Output is correct |
5 |
Correct |
31 ms |
65748 KB |
Output is correct |
6 |
Correct |
30 ms |
65744 KB |
Output is correct |
7 |
Correct |
30 ms |
65680 KB |
Output is correct |
8 |
Correct |
29 ms |
65748 KB |
Output is correct |
9 |
Correct |
29 ms |
65748 KB |
Output is correct |
10 |
Correct |
5 ms |
10452 KB |
Output is correct |
11 |
Correct |
5 ms |
10452 KB |
Output is correct |
12 |
Correct |
5 ms |
10452 KB |
Output is correct |
13 |
Correct |
2 ms |
2644 KB |
Output is correct |
14 |
Correct |
2 ms |
2644 KB |
Output is correct |
15 |
Correct |
2 ms |
2632 KB |
Output is correct |
16 |
Correct |
28 ms |
65740 KB |
Output is correct |
17 |
Correct |
30 ms |
65748 KB |
Output is correct |
18 |
Correct |
29 ms |
65748 KB |
Output is correct |
19 |
Correct |
1 ms |
2644 KB |
Output is correct |
20 |
Correct |
2 ms |
2644 KB |
Output is correct |
21 |
Correct |
2 ms |
2644 KB |
Output is correct |
22 |
Correct |
1 ms |
2644 KB |
Output is correct |
23 |
Correct |
217 ms |
66016 KB |
Output is correct |
24 |
Correct |
551 ms |
66008 KB |
Output is correct |
25 |
Correct |
400 ms |
66044 KB |
Output is correct |
26 |
Correct |
419 ms |
65932 KB |
Output is correct |
27 |
Correct |
421 ms |
65996 KB |
Output is correct |
28 |
Correct |
283 ms |
65956 KB |
Output is correct |
29 |
Correct |
470 ms |
66028 KB |
Output is correct |
30 |
Correct |
375 ms |
65876 KB |
Output is correct |
31 |
Correct |
372 ms |
66004 KB |
Output is correct |
32 |
Correct |
275 ms |
65948 KB |
Output is correct |
33 |
Correct |
6 ms |
10836 KB |
Output is correct |
34 |
Correct |
6 ms |
10868 KB |
Output is correct |
35 |
Correct |
6 ms |
10836 KB |
Output is correct |
36 |
Correct |
2 ms |
2772 KB |
Output is correct |
37 |
Correct |
2 ms |
2772 KB |
Output is correct |
38 |
Correct |
3 ms |
2772 KB |
Output is correct |
39 |
Correct |
26 ms |
65708 KB |
Output is correct |
40 |
Correct |
26 ms |
65748 KB |
Output is correct |
41 |
Correct |
29 ms |
65716 KB |
Output is correct |
42 |
Correct |
31 ms |
65784 KB |
Output is correct |
43 |
Correct |
28 ms |
65740 KB |
Output is correct |
44 |
Correct |
29 ms |
65748 KB |
Output is correct |
45 |
Correct |
30 ms |
65720 KB |
Output is correct |
46 |
Correct |
29 ms |
65692 KB |
Output is correct |
47 |
Correct |
5 ms |
10452 KB |
Output is correct |
48 |
Correct |
5 ms |
10452 KB |
Output is correct |
49 |
Correct |
5 ms |
10452 KB |
Output is correct |
50 |
Correct |
1 ms |
2644 KB |
Output is correct |
51 |
Correct |
2 ms |
2644 KB |
Output is correct |
52 |
Correct |
2 ms |
2644 KB |
Output is correct |
53 |
Correct |
339 ms |
66044 KB |
Output is correct |
54 |
Correct |
548 ms |
65964 KB |
Output is correct |
55 |
Correct |
349 ms |
65996 KB |
Output is correct |
56 |
Correct |
3 ms |
2772 KB |
Output is correct |
57 |
Correct |
3 ms |
2772 KB |
Output is correct |
58 |
Correct |
27 ms |
65748 KB |
Output is correct |
59 |
Correct |
31 ms |
65732 KB |
Output is correct |
60 |
Correct |
30 ms |
65868 KB |
Output is correct |
61 |
Correct |
2 ms |
2644 KB |
Output is correct |
62 |
Correct |
2 ms |
2644 KB |
Output is correct |
63 |
Correct |
2 ms |
2644 KB |
Output is correct |
64 |
Correct |
515 ms |
65960 KB |
Output is correct |
65 |
Correct |
542 ms |
65952 KB |
Output is correct |
66 |
Correct |
330 ms |
65996 KB |
Output is correct |
67 |
Correct |
289 ms |
66140 KB |
Output is correct |
68 |
Correct |
302 ms |
65924 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
48 ms |
20128 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
48 ms |
20128 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
2644 KB |
Output is correct |
2 |
Correct |
2 ms |
2772 KB |
Output is correct |
3 |
Correct |
2 ms |
2772 KB |
Output is correct |
4 |
Correct |
3 ms |
2772 KB |
Output is correct |
5 |
Correct |
2 ms |
2644 KB |
Output is correct |
6 |
Correct |
1 ms |
2644 KB |
Output is correct |
7 |
Correct |
2 ms |
2644 KB |
Output is correct |
8 |
Correct |
3 ms |
2772 KB |
Output is correct |
9 |
Correct |
2 ms |
2772 KB |
Output is correct |
10 |
Correct |
1 ms |
2644 KB |
Output is correct |
11 |
Correct |
2 ms |
2644 KB |
Output is correct |
12 |
Correct |
29 ms |
7848 KB |
Output is correct |
13 |
Correct |
51 ms |
11452 KB |
Output is correct |
14 |
Correct |
54 ms |
10904 KB |
Output is correct |
15 |
Correct |
58 ms |
11616 KB |
Output is correct |
16 |
Correct |
60 ms |
11812 KB |
Output is correct |
17 |
Correct |
58 ms |
11784 KB |
Output is correct |
18 |
Correct |
2 ms |
2644 KB |
Output is correct |
19 |
Correct |
48 ms |
10708 KB |
Output is correct |
20 |
Correct |
2 ms |
2644 KB |
Output is correct |
21 |
Correct |
69 ms |
31664 KB |
Output is correct |
22 |
Correct |
75 ms |
36256 KB |
Output is correct |
23 |
Correct |
77 ms |
37960 KB |
Output is correct |
24 |
Correct |
77 ms |
38088 KB |
Output is correct |
25 |
Correct |
6 ms |
10964 KB |
Output is correct |
26 |
Correct |
6 ms |
10996 KB |
Output is correct |
27 |
Correct |
6 ms |
10964 KB |
Output is correct |
28 |
Correct |
5 ms |
10580 KB |
Output is correct |
29 |
Correct |
5 ms |
10452 KB |
Output is correct |
30 |
Correct |
5 ms |
10452 KB |
Output is correct |
31 |
Correct |
44 ms |
26468 KB |
Output is correct |
32 |
Correct |
72 ms |
37244 KB |
Output is correct |
33 |
Correct |
1 ms |
2644 KB |
Output is correct |
34 |
Correct |
66 ms |
34676 KB |
Output is correct |
35 |
Correct |
71 ms |
37180 KB |
Output is correct |
36 |
Correct |
1 ms |
2644 KB |
Output is correct |
37 |
Correct |
26 ms |
65688 KB |
Output is correct |
38 |
Correct |
25 ms |
65756 KB |
Output is correct |
39 |
Correct |
28 ms |
65672 KB |
Output is correct |
40 |
Correct |
31 ms |
65748 KB |
Output is correct |
41 |
Correct |
30 ms |
65744 KB |
Output is correct |
42 |
Correct |
30 ms |
65680 KB |
Output is correct |
43 |
Correct |
29 ms |
65748 KB |
Output is correct |
44 |
Correct |
29 ms |
65748 KB |
Output is correct |
45 |
Correct |
5 ms |
10452 KB |
Output is correct |
46 |
Correct |
5 ms |
10452 KB |
Output is correct |
47 |
Correct |
5 ms |
10452 KB |
Output is correct |
48 |
Correct |
2 ms |
2644 KB |
Output is correct |
49 |
Correct |
2 ms |
2644 KB |
Output is correct |
50 |
Correct |
2 ms |
2632 KB |
Output is correct |
51 |
Correct |
28 ms |
65740 KB |
Output is correct |
52 |
Correct |
30 ms |
65748 KB |
Output is correct |
53 |
Correct |
29 ms |
65748 KB |
Output is correct |
54 |
Correct |
1 ms |
2644 KB |
Output is correct |
55 |
Correct |
2 ms |
2644 KB |
Output is correct |
56 |
Correct |
2 ms |
2644 KB |
Output is correct |
57 |
Correct |
1 ms |
2644 KB |
Output is correct |
58 |
Correct |
217 ms |
66016 KB |
Output is correct |
59 |
Correct |
551 ms |
66008 KB |
Output is correct |
60 |
Correct |
400 ms |
66044 KB |
Output is correct |
61 |
Correct |
419 ms |
65932 KB |
Output is correct |
62 |
Correct |
421 ms |
65996 KB |
Output is correct |
63 |
Correct |
283 ms |
65956 KB |
Output is correct |
64 |
Correct |
470 ms |
66028 KB |
Output is correct |
65 |
Correct |
375 ms |
65876 KB |
Output is correct |
66 |
Correct |
372 ms |
66004 KB |
Output is correct |
67 |
Correct |
275 ms |
65948 KB |
Output is correct |
68 |
Correct |
6 ms |
10836 KB |
Output is correct |
69 |
Correct |
6 ms |
10868 KB |
Output is correct |
70 |
Correct |
6 ms |
10836 KB |
Output is correct |
71 |
Correct |
2 ms |
2772 KB |
Output is correct |
72 |
Correct |
2 ms |
2772 KB |
Output is correct |
73 |
Correct |
3 ms |
2772 KB |
Output is correct |
74 |
Correct |
26 ms |
65708 KB |
Output is correct |
75 |
Correct |
26 ms |
65748 KB |
Output is correct |
76 |
Correct |
29 ms |
65716 KB |
Output is correct |
77 |
Correct |
31 ms |
65784 KB |
Output is correct |
78 |
Correct |
28 ms |
65740 KB |
Output is correct |
79 |
Correct |
29 ms |
65748 KB |
Output is correct |
80 |
Correct |
30 ms |
65720 KB |
Output is correct |
81 |
Correct |
29 ms |
65692 KB |
Output is correct |
82 |
Correct |
5 ms |
10452 KB |
Output is correct |
83 |
Correct |
5 ms |
10452 KB |
Output is correct |
84 |
Correct |
5 ms |
10452 KB |
Output is correct |
85 |
Correct |
1 ms |
2644 KB |
Output is correct |
86 |
Correct |
2 ms |
2644 KB |
Output is correct |
87 |
Correct |
2 ms |
2644 KB |
Output is correct |
88 |
Correct |
339 ms |
66044 KB |
Output is correct |
89 |
Correct |
548 ms |
65964 KB |
Output is correct |
90 |
Correct |
349 ms |
65996 KB |
Output is correct |
91 |
Correct |
3 ms |
2772 KB |
Output is correct |
92 |
Correct |
3 ms |
2772 KB |
Output is correct |
93 |
Correct |
27 ms |
65748 KB |
Output is correct |
94 |
Correct |
31 ms |
65732 KB |
Output is correct |
95 |
Correct |
30 ms |
65868 KB |
Output is correct |
96 |
Correct |
2 ms |
2644 KB |
Output is correct |
97 |
Correct |
2 ms |
2644 KB |
Output is correct |
98 |
Correct |
2 ms |
2644 KB |
Output is correct |
99 |
Correct |
515 ms |
65960 KB |
Output is correct |
100 |
Correct |
542 ms |
65952 KB |
Output is correct |
101 |
Correct |
330 ms |
65996 KB |
Output is correct |
102 |
Correct |
289 ms |
66140 KB |
Output is correct |
103 |
Correct |
302 ms |
65924 KB |
Output is correct |
104 |
Runtime error |
48 ms |
20128 KB |
Execution killed with signal 6 |
105 |
Halted |
0 ms |
0 KB |
- |