#include "roads.h"
#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define sz(a) (int)(a.size())
#define all(a) a.begin(),a.end()
#define lb lower_bound
#define ub upper_bound
#define owo ios_base::sync_with_stdio(0);cin.tie(0);
#define INF (ll)(1e18)
#define debug(...) fprintf(stderr, __VA_ARGS__),fflush(stderr)
#define time__(d) for(long blockTime = 0; (blockTime == 0 ? (blockTime=clock()) != 0 : false);\
debug("%s time : %.4fs\n", d, (double)(clock() - blockTime) / CLOCKS_PER_SEC))
typedef long long int ll;
typedef long double ld;
typedef pair<ll,ll> PII;
typedef pair<int,int> pii;
typedef vector<vector<int>> vii;
typedef vector<vector<ll>> VII;
ll gcd(ll a,ll b){if(!b)return a;else return gcd(b,a%b);}
vector<vector<PII>> adj(200005);
ll dp[2][2001];
void dfs(int v,int u,int k,ll w){
int need = sz(adj[v])-k;
need = max(need,0);
vector<PII>choose;
vector<ll>rest;
for(PII z:adj[v]){
int x = z.fi;
ll w = z.se;
if(x != u){
dfs(x,v,k,w);
if(dp[0][x] == INF){
dp[0][v]+=dp[1][x];
dp[1][v]+=dp[1][x];
need--;
}
else choose.pb({dp[1][x]-min(dp[1][x],dp[0][x]),x});
}
}
sort(all(choose));
if(need <= sz(choose)){
for(int i=0;i<need;i++)dp[0][v]+=dp[1][choose[i].se];
for(int i=need;i<sz(choose);i++)dp[0][v]+=min(dp[0][choose[i].se],dp[1][choose[i].se]);
}
else dp[0][v] = INF;
if(u != -1){
dp[1][v] += w;
need = max(0,need-1);
rest.clear();
for(int i=0;i<need;i++)dp[1][v]+=dp[1][choose[i].se];
for(int i=need;i<sz(choose);i++)dp[1][v]+=min(dp[0][choose[i].se],dp[1][choose[i].se]);
}
}
vector<ll>ans;
vector<ll> minimum_closure_costs(int n, vector<int> u,vector<int>v,vector<int>w){
bool sub1 =1;
bool sub2 = 1;
bool sub4 = 1;
ans.resize(n);
for(int i=0;i<n-1;i++){
ans[0]+=w[i];
if(u[i] != i || v[i] != i+1)sub2 = 0;
if(u[i])sub1 = 0;
adj[u[i]].pb({v[i],w[i]});
adj[v[i]].pb({u[i],w[i]});
}
if(sub1){
sort(all(w));
ll sum = 0;
for(int i=0;i<n-1;i++){
sum+=w[i];
ans[n-i-2] = sum;
}
return ans;
}
if(sub2){
VII d(2,vector<ll>(n,INF));
d[0][0] = 0;
d[1][0] = w[0];
for(int i=1;i<n-1;i++){
d[0][i] = d[1][i-1];
d[1][i] = min(d[0][i-1],d[1][i-1])+w[i];
}
ans[1] = min(d[0][n-2],d[1][n-2]);
return ans;
}
for(int i=0;i<n;i++){
for(int i=0;i<n;i++)dp[0][i] = dp[1][i] = 0;
dfs(0,-1,i,0);
//if(i==1)for(int j=0;j<n;j++)cout<<dp[0][j]<<" "<<dp[1][j]<<'\n';
ans[i] = dp[0][0];
}
return ans;
}
Compilation message
roads.cpp: In function 'std::vector<long long int> minimum_closure_costs(int, std::vector<int>, std::vector<int>, std::vector<int>)':
roads.cpp:62:7: warning: unused variable 'sub4' [-Wunused-variable]
62 | bool sub4 = 1;
| ^~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
4940 KB |
Output is correct |
2 |
Correct |
7 ms |
5188 KB |
Output is correct |
3 |
Correct |
7 ms |
5168 KB |
Output is correct |
4 |
Correct |
5 ms |
5196 KB |
Output is correct |
5 |
Correct |
4 ms |
4940 KB |
Output is correct |
6 |
Correct |
8 ms |
4972 KB |
Output is correct |
7 |
Correct |
3 ms |
5008 KB |
Output is correct |
8 |
Correct |
12 ms |
5068 KB |
Output is correct |
9 |
Correct |
7 ms |
5196 KB |
Output is correct |
10 |
Correct |
11 ms |
5008 KB |
Output is correct |
11 |
Correct |
4 ms |
4940 KB |
Output is correct |
12 |
Correct |
47 ms |
10648 KB |
Output is correct |
13 |
Correct |
72 ms |
14536 KB |
Output is correct |
14 |
Correct |
79 ms |
14420 KB |
Output is correct |
15 |
Correct |
84 ms |
15444 KB |
Output is correct |
16 |
Correct |
95 ms |
15584 KB |
Output is correct |
17 |
Correct |
96 ms |
15704 KB |
Output is correct |
18 |
Correct |
4 ms |
4904 KB |
Output is correct |
19 |
Correct |
79 ms |
13580 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
4 ms |
4940 KB |
Output is correct |
2 |
Correct |
84 ms |
15180 KB |
Output is correct |
3 |
Correct |
64 ms |
16484 KB |
Output is correct |
4 |
Correct |
84 ms |
17304 KB |
Output is correct |
5 |
Correct |
72 ms |
17296 KB |
Output is correct |
6 |
Correct |
5 ms |
5144 KB |
Output is correct |
7 |
Correct |
7 ms |
5144 KB |
Output is correct |
8 |
Correct |
6 ms |
5180 KB |
Output is correct |
9 |
Correct |
4 ms |
4940 KB |
Output is correct |
10 |
Correct |
4 ms |
5004 KB |
Output is correct |
11 |
Correct |
5 ms |
4940 KB |
Output is correct |
12 |
Correct |
39 ms |
11832 KB |
Output is correct |
13 |
Correct |
59 ms |
16480 KB |
Output is correct |
14 |
Correct |
3 ms |
4940 KB |
Output is correct |
15 |
Correct |
53 ms |
15316 KB |
Output is correct |
16 |
Correct |
76 ms |
16512 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
5000 KB |
Output is correct |
2 |
Correct |
4 ms |
5004 KB |
Output is correct |
3 |
Correct |
4 ms |
5012 KB |
Output is correct |
4 |
Correct |
8 ms |
5004 KB |
Output is correct |
5 |
Correct |
8 ms |
4940 KB |
Output is correct |
6 |
Correct |
7 ms |
4940 KB |
Output is correct |
7 |
Correct |
7 ms |
4940 KB |
Output is correct |
8 |
Correct |
7 ms |
4940 KB |
Output is correct |
9 |
Correct |
8 ms |
4940 KB |
Output is correct |
10 |
Correct |
12 ms |
4940 KB |
Output is correct |
11 |
Correct |
4 ms |
4940 KB |
Output is correct |
12 |
Correct |
5 ms |
4940 KB |
Output is correct |
13 |
Correct |
4 ms |
4940 KB |
Output is correct |
14 |
Correct |
4 ms |
5004 KB |
Output is correct |
15 |
Correct |
4 ms |
4940 KB |
Output is correct |
16 |
Correct |
5 ms |
4940 KB |
Output is correct |
17 |
Correct |
7 ms |
5028 KB |
Output is correct |
18 |
Correct |
7 ms |
4984 KB |
Output is correct |
19 |
Correct |
5 ms |
5000 KB |
Output is correct |
20 |
Correct |
4 ms |
4940 KB |
Output is correct |
21 |
Correct |
4 ms |
5000 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
5000 KB |
Output is correct |
2 |
Correct |
4 ms |
5004 KB |
Output is correct |
3 |
Correct |
4 ms |
5012 KB |
Output is correct |
4 |
Correct |
8 ms |
5004 KB |
Output is correct |
5 |
Correct |
8 ms |
4940 KB |
Output is correct |
6 |
Correct |
7 ms |
4940 KB |
Output is correct |
7 |
Correct |
7 ms |
4940 KB |
Output is correct |
8 |
Correct |
7 ms |
4940 KB |
Output is correct |
9 |
Correct |
8 ms |
4940 KB |
Output is correct |
10 |
Correct |
12 ms |
4940 KB |
Output is correct |
11 |
Correct |
4 ms |
4940 KB |
Output is correct |
12 |
Correct |
5 ms |
4940 KB |
Output is correct |
13 |
Correct |
4 ms |
4940 KB |
Output is correct |
14 |
Correct |
4 ms |
5004 KB |
Output is correct |
15 |
Correct |
4 ms |
4940 KB |
Output is correct |
16 |
Correct |
5 ms |
4940 KB |
Output is correct |
17 |
Correct |
7 ms |
5028 KB |
Output is correct |
18 |
Correct |
7 ms |
4984 KB |
Output is correct |
19 |
Correct |
5 ms |
5000 KB |
Output is correct |
20 |
Correct |
4 ms |
4940 KB |
Output is correct |
21 |
Correct |
4 ms |
5000 KB |
Output is correct |
22 |
Correct |
4 ms |
5016 KB |
Output is correct |
23 |
Correct |
164 ms |
5168 KB |
Output is correct |
24 |
Correct |
497 ms |
5252 KB |
Output is correct |
25 |
Correct |
344 ms |
5196 KB |
Output is correct |
26 |
Correct |
311 ms |
5216 KB |
Output is correct |
27 |
Correct |
339 ms |
5264 KB |
Output is correct |
28 |
Correct |
397 ms |
5252 KB |
Output is correct |
29 |
Correct |
308 ms |
5204 KB |
Output is correct |
30 |
Correct |
243 ms |
5244 KB |
Output is correct |
31 |
Correct |
291 ms |
5228 KB |
Output is correct |
32 |
Correct |
295 ms |
5232 KB |
Output is correct |
33 |
Correct |
6 ms |
5196 KB |
Output is correct |
34 |
Correct |
5 ms |
5196 KB |
Output is correct |
35 |
Correct |
7 ms |
5144 KB |
Output is correct |
36 |
Correct |
8 ms |
5196 KB |
Output is correct |
37 |
Correct |
6 ms |
5196 KB |
Output is correct |
38 |
Correct |
5 ms |
5232 KB |
Output is correct |
39 |
Correct |
5 ms |
4940 KB |
Output is correct |
40 |
Correct |
5 ms |
5000 KB |
Output is correct |
41 |
Correct |
6 ms |
4940 KB |
Output is correct |
42 |
Correct |
7 ms |
5004 KB |
Output is correct |
43 |
Correct |
7 ms |
4940 KB |
Output is correct |
44 |
Correct |
6 ms |
5004 KB |
Output is correct |
45 |
Correct |
6 ms |
5004 KB |
Output is correct |
46 |
Correct |
6 ms |
4940 KB |
Output is correct |
47 |
Correct |
6 ms |
4992 KB |
Output is correct |
48 |
Correct |
5 ms |
5004 KB |
Output is correct |
49 |
Correct |
15 ms |
4964 KB |
Output is correct |
50 |
Correct |
5 ms |
4940 KB |
Output is correct |
51 |
Correct |
4 ms |
5004 KB |
Output is correct |
52 |
Correct |
4 ms |
5004 KB |
Output is correct |
53 |
Correct |
199 ms |
5068 KB |
Output is correct |
54 |
Correct |
359 ms |
5228 KB |
Output is correct |
55 |
Correct |
347 ms |
5244 KB |
Output is correct |
56 |
Correct |
5 ms |
5144 KB |
Output is correct |
57 |
Correct |
6 ms |
5196 KB |
Output is correct |
58 |
Correct |
5 ms |
4940 KB |
Output is correct |
59 |
Correct |
7 ms |
5004 KB |
Output is correct |
60 |
Correct |
7 ms |
4940 KB |
Output is correct |
61 |
Correct |
4 ms |
4940 KB |
Output is correct |
62 |
Correct |
5 ms |
5008 KB |
Output is correct |
63 |
Correct |
4 ms |
4940 KB |
Output is correct |
64 |
Correct |
318 ms |
5216 KB |
Output is correct |
65 |
Correct |
331 ms |
5196 KB |
Output is correct |
66 |
Correct |
257 ms |
5212 KB |
Output is correct |
67 |
Correct |
319 ms |
5216 KB |
Output is correct |
68 |
Correct |
253 ms |
5212 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
71 ms |
27752 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
71 ms |
27752 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
5 ms |
4940 KB |
Output is correct |
2 |
Correct |
7 ms |
5188 KB |
Output is correct |
3 |
Correct |
7 ms |
5168 KB |
Output is correct |
4 |
Correct |
5 ms |
5196 KB |
Output is correct |
5 |
Correct |
4 ms |
4940 KB |
Output is correct |
6 |
Correct |
8 ms |
4972 KB |
Output is correct |
7 |
Correct |
3 ms |
5008 KB |
Output is correct |
8 |
Correct |
12 ms |
5068 KB |
Output is correct |
9 |
Correct |
7 ms |
5196 KB |
Output is correct |
10 |
Correct |
11 ms |
5008 KB |
Output is correct |
11 |
Correct |
4 ms |
4940 KB |
Output is correct |
12 |
Correct |
47 ms |
10648 KB |
Output is correct |
13 |
Correct |
72 ms |
14536 KB |
Output is correct |
14 |
Correct |
79 ms |
14420 KB |
Output is correct |
15 |
Correct |
84 ms |
15444 KB |
Output is correct |
16 |
Correct |
95 ms |
15584 KB |
Output is correct |
17 |
Correct |
96 ms |
15704 KB |
Output is correct |
18 |
Correct |
4 ms |
4904 KB |
Output is correct |
19 |
Correct |
79 ms |
13580 KB |
Output is correct |
20 |
Correct |
4 ms |
4940 KB |
Output is correct |
21 |
Correct |
84 ms |
15180 KB |
Output is correct |
22 |
Correct |
64 ms |
16484 KB |
Output is correct |
23 |
Correct |
84 ms |
17304 KB |
Output is correct |
24 |
Correct |
72 ms |
17296 KB |
Output is correct |
25 |
Correct |
5 ms |
5144 KB |
Output is correct |
26 |
Correct |
7 ms |
5144 KB |
Output is correct |
27 |
Correct |
6 ms |
5180 KB |
Output is correct |
28 |
Correct |
4 ms |
4940 KB |
Output is correct |
29 |
Correct |
4 ms |
5004 KB |
Output is correct |
30 |
Correct |
5 ms |
4940 KB |
Output is correct |
31 |
Correct |
39 ms |
11832 KB |
Output is correct |
32 |
Correct |
59 ms |
16480 KB |
Output is correct |
33 |
Correct |
3 ms |
4940 KB |
Output is correct |
34 |
Correct |
53 ms |
15316 KB |
Output is correct |
35 |
Correct |
76 ms |
16512 KB |
Output is correct |
36 |
Correct |
5 ms |
5000 KB |
Output is correct |
37 |
Correct |
4 ms |
5004 KB |
Output is correct |
38 |
Correct |
4 ms |
5012 KB |
Output is correct |
39 |
Correct |
8 ms |
5004 KB |
Output is correct |
40 |
Correct |
8 ms |
4940 KB |
Output is correct |
41 |
Correct |
7 ms |
4940 KB |
Output is correct |
42 |
Correct |
7 ms |
4940 KB |
Output is correct |
43 |
Correct |
7 ms |
4940 KB |
Output is correct |
44 |
Correct |
8 ms |
4940 KB |
Output is correct |
45 |
Correct |
12 ms |
4940 KB |
Output is correct |
46 |
Correct |
4 ms |
4940 KB |
Output is correct |
47 |
Correct |
5 ms |
4940 KB |
Output is correct |
48 |
Correct |
4 ms |
4940 KB |
Output is correct |
49 |
Correct |
4 ms |
5004 KB |
Output is correct |
50 |
Correct |
4 ms |
4940 KB |
Output is correct |
51 |
Correct |
5 ms |
4940 KB |
Output is correct |
52 |
Correct |
7 ms |
5028 KB |
Output is correct |
53 |
Correct |
7 ms |
4984 KB |
Output is correct |
54 |
Correct |
5 ms |
5000 KB |
Output is correct |
55 |
Correct |
4 ms |
4940 KB |
Output is correct |
56 |
Correct |
4 ms |
5000 KB |
Output is correct |
57 |
Correct |
4 ms |
5016 KB |
Output is correct |
58 |
Correct |
164 ms |
5168 KB |
Output is correct |
59 |
Correct |
497 ms |
5252 KB |
Output is correct |
60 |
Correct |
344 ms |
5196 KB |
Output is correct |
61 |
Correct |
311 ms |
5216 KB |
Output is correct |
62 |
Correct |
339 ms |
5264 KB |
Output is correct |
63 |
Correct |
397 ms |
5252 KB |
Output is correct |
64 |
Correct |
308 ms |
5204 KB |
Output is correct |
65 |
Correct |
243 ms |
5244 KB |
Output is correct |
66 |
Correct |
291 ms |
5228 KB |
Output is correct |
67 |
Correct |
295 ms |
5232 KB |
Output is correct |
68 |
Correct |
6 ms |
5196 KB |
Output is correct |
69 |
Correct |
5 ms |
5196 KB |
Output is correct |
70 |
Correct |
7 ms |
5144 KB |
Output is correct |
71 |
Correct |
8 ms |
5196 KB |
Output is correct |
72 |
Correct |
6 ms |
5196 KB |
Output is correct |
73 |
Correct |
5 ms |
5232 KB |
Output is correct |
74 |
Correct |
5 ms |
4940 KB |
Output is correct |
75 |
Correct |
5 ms |
5000 KB |
Output is correct |
76 |
Correct |
6 ms |
4940 KB |
Output is correct |
77 |
Correct |
7 ms |
5004 KB |
Output is correct |
78 |
Correct |
7 ms |
4940 KB |
Output is correct |
79 |
Correct |
6 ms |
5004 KB |
Output is correct |
80 |
Correct |
6 ms |
5004 KB |
Output is correct |
81 |
Correct |
6 ms |
4940 KB |
Output is correct |
82 |
Correct |
6 ms |
4992 KB |
Output is correct |
83 |
Correct |
5 ms |
5004 KB |
Output is correct |
84 |
Correct |
15 ms |
4964 KB |
Output is correct |
85 |
Correct |
5 ms |
4940 KB |
Output is correct |
86 |
Correct |
4 ms |
5004 KB |
Output is correct |
87 |
Correct |
4 ms |
5004 KB |
Output is correct |
88 |
Correct |
199 ms |
5068 KB |
Output is correct |
89 |
Correct |
359 ms |
5228 KB |
Output is correct |
90 |
Correct |
347 ms |
5244 KB |
Output is correct |
91 |
Correct |
5 ms |
5144 KB |
Output is correct |
92 |
Correct |
6 ms |
5196 KB |
Output is correct |
93 |
Correct |
5 ms |
4940 KB |
Output is correct |
94 |
Correct |
7 ms |
5004 KB |
Output is correct |
95 |
Correct |
7 ms |
4940 KB |
Output is correct |
96 |
Correct |
4 ms |
4940 KB |
Output is correct |
97 |
Correct |
5 ms |
5008 KB |
Output is correct |
98 |
Correct |
4 ms |
4940 KB |
Output is correct |
99 |
Correct |
318 ms |
5216 KB |
Output is correct |
100 |
Correct |
331 ms |
5196 KB |
Output is correct |
101 |
Correct |
257 ms |
5212 KB |
Output is correct |
102 |
Correct |
319 ms |
5216 KB |
Output is correct |
103 |
Correct |
253 ms |
5212 KB |
Output is correct |
104 |
Runtime error |
71 ms |
27752 KB |
Execution killed with signal 11 |
105 |
Halted |
0 ms |
0 KB |
- |