#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);}
const int MAXN = 2e5+5;
vector<vector<pii>> g1(MAXN),g2(MAXN); //g1 is original graph,g2 is subgraph after we deleted nodes with degree<=k
vector<int>par(MAXN),szz(MAXN,1),act(MAXN); //act means whether a node is active / has degree>k
vector<ll>ans;
VII dp(2,vector<ll>(MAXN,0));
int find(int x){
if(x == par[x])return x;
else return par[x] = find(par[x]);
}
void unite(int a,int b){
a = find(a);
b = find(b);
if(a==b)return;
if(szz[a] > szz[b])swap(a,b);
par[a] = b;
szz[b]+=szz[a];
}
struct node{ //dynamic segtree
ll sum,cnt;
ll L,R;
node *lf=nullptr,*rg = nullptr;
node(){}
node(ll _L,ll _R){
L = _L;
R = _R;
}
void upd(ll pos,ll c){
if(L==R){
sum+=c*pos;
cnt+=c;
return;
}
ll tm = (L+R)/2;
if(pos<=tm){
if(lf == nullptr)lf = new node(L,tm);
lf->upd(pos,c);
}else{
if(rg == nullptr)rg = new node(tm+1,R);
rg->upd(pos,c);
}
sum=cnt=0;
if(lf != nullptr){
sum+=lf->sum;
cnt+=lf->cnt;
}
if(rg != nullptr){
sum+=rg->sum;
cnt+=rg->cnt;
}
}
ll query(ll v){
//cout<<L<<" "<<R<<" "<<cnt<<" "<<v<<'\n';
if(L == R)return v*L;
if(cnt == v)return sum;
ll tm = (L+R)/2;
if(lf == nullptr)lf = new node(L,tm);
if(lf->cnt >= v)return lf->query(v);
else{
if(rg == nullptr)rg = new node(tm+1,R);
return lf->sum + rg->query(v-lf->cnt);
}
}
};
vector<node>seg(MAXN);
void dfs(int v,int u,ll w,int k){
int need = sz(g1[v])-k;
ll cur = 0;
vector<ll>choice;
for(auto x:g2[v]){
if(x.fi != u){
dfs(x.fi,v,x.se,k);
if(dp[1][x.fi] <= dp[0][x.fi]){ //we will take this edge anyways
cur+=dp[1][x.fi];
need--;
}else{
cur+=dp[0][x.fi];
choice.pb(dp[1][x.fi]-dp[0][x.fi]);
}
}
}
dp[0][v] = dp[1][v] = INF;
sort(all(choice));
if(need<=0)dp[0][v] = cur;
if(need-1<=0)dp[1][v] = w+cur;
if(seg[v].cnt >= need)dp[0][v] = cur + (need>0?seg[v].query(need):0);
if(seg[v].cnt >= need-1)dp[1][v] = cur + w + ((need-1)>0?seg[v].query(need-1):0); //take extra edges only from inactive
ll tmp = cur+w;
for(int i=0;i<min(sz(choice),need);i++){
cur+=choice[i];
int take = need-i-1;
if(seg[v].cnt>=take)dp[0][v] = min(dp[0][v],cur + (take>0?seg[v].query(take):0));
}
need--;
for(int i=0;i<min(sz(choice),need);i++){
tmp+=choice[i];
int take = need-i-1;
if(seg[v].cnt>=take)dp[1][v] = min(dp[1][v],tmp + (take>0?seg[v].query(take):0));
}
}
vector<ll> minimum_closure_costs(int n, vector<int> u,vector<int>v,vector<int>w){
ans.resize(n);
for(int i=0;i<n-1;i++){
g1[u[i]].pb({v[i],w[i]});
g1[v[i]].pb({u[i],w[i]});
}
for(int i=0;i<n;i++){
seg[i] = node(1,1e9);
for(auto x:g1[i])seg[i].upd(x.se,1); //currently graph is empty so we add every single edge as "inactive edge"
}
vector<int>c(n);
for(int i=0;i<n;i++){c[i] = i;par[i] = i;}
sort(all(c),[&](int x,int y){return sz(g1[x]) > sz(g1[y]);});
//sort vertex by degree and we sweep
int ptr = -1;
vector<int>vis(n);
for(int i=n-1;i>=0;i--){
while(ptr+1<n && sz(g1[c[ptr+1]]) > i){
ptr++;
int d = c[ptr];
act[d] = 1;
for(auto x:g1[d]){
if(act[x.fi]){
seg[x.fi].upd(x.se,-1); //these edges become active now,remove them from segtree
seg[d].upd(x.se,-1);
g2[x.fi].pb({d,x.se});
g2[d].pb({x.fi,x.se});
unite(x.fi,d);
}
}
}
vector<int>meet;
for(int j=0;j<=ptr;j++){
int a = find(c[j]);
if(vis[a])continue;
else{
dfs(a,-1,0,i);
meet.pb(a);
vis[a] = 1;
ans[i]+=dp[0][a];
}
}
for(int x:meet)vis[x] = 0;
}
return ans;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
16 ms |
24496 KB |
Output is correct |
2 |
Correct |
29 ms |
30640 KB |
Output is correct |
3 |
Correct |
28 ms |
31024 KB |
Output is correct |
4 |
Correct |
22 ms |
28464 KB |
Output is correct |
5 |
Correct |
16 ms |
25008 KB |
Output is correct |
6 |
Correct |
17 ms |
25264 KB |
Output is correct |
7 |
Correct |
16 ms |
24960 KB |
Output is correct |
8 |
Correct |
23 ms |
27976 KB |
Output is correct |
9 |
Correct |
22 ms |
28592 KB |
Output is correct |
10 |
Correct |
20 ms |
24860 KB |
Output is correct |
11 |
Correct |
20 ms |
24984 KB |
Output is correct |
12 |
Correct |
241 ms |
145384 KB |
Output is correct |
13 |
Correct |
406 ms |
225536 KB |
Output is correct |
14 |
Correct |
650 ms |
280352 KB |
Output is correct |
15 |
Correct |
756 ms |
313712 KB |
Output is correct |
16 |
Correct |
798 ms |
313948 KB |
Output is correct |
17 |
Correct |
434 ms |
220120 KB |
Output is correct |
18 |
Correct |
15 ms |
24496 KB |
Output is correct |
19 |
Correct |
358 ms |
206208 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
16 ms |
24496 KB |
Output is correct |
2 |
Correct |
548 ms |
345652 KB |
Output is correct |
3 |
Correct |
642 ms |
392296 KB |
Output is correct |
4 |
Correct |
680 ms |
415620 KB |
Output is correct |
5 |
Correct |
407 ms |
230784 KB |
Output is correct |
6 |
Correct |
28 ms |
31228 KB |
Output is correct |
7 |
Correct |
34 ms |
32296 KB |
Output is correct |
8 |
Correct |
20 ms |
28204 KB |
Output is correct |
9 |
Correct |
16 ms |
25176 KB |
Output is correct |
10 |
Correct |
18 ms |
25212 KB |
Output is correct |
11 |
Correct |
17 ms |
24988 KB |
Output is correct |
12 |
Correct |
224 ms |
161068 KB |
Output is correct |
13 |
Correct |
424 ms |
253764 KB |
Output is correct |
14 |
Correct |
15 ms |
24496 KB |
Output is correct |
15 |
Correct |
348 ms |
215940 KB |
Output is correct |
16 |
Correct |
418 ms |
237056 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
15 ms |
24496 KB |
Output is correct |
2 |
Correct |
15 ms |
24524 KB |
Output is correct |
3 |
Correct |
14 ms |
24588 KB |
Output is correct |
4 |
Correct |
16 ms |
25112 KB |
Output is correct |
5 |
Correct |
20 ms |
25316 KB |
Output is correct |
6 |
Correct |
17 ms |
24856 KB |
Output is correct |
7 |
Correct |
16 ms |
25216 KB |
Output is correct |
8 |
Correct |
16 ms |
24976 KB |
Output is correct |
9 |
Correct |
16 ms |
25280 KB |
Output is correct |
10 |
Correct |
16 ms |
25156 KB |
Output is correct |
11 |
Correct |
16 ms |
25228 KB |
Output is correct |
12 |
Correct |
15 ms |
24880 KB |
Output is correct |
13 |
Correct |
16 ms |
24932 KB |
Output is correct |
14 |
Correct |
17 ms |
25264 KB |
Output is correct |
15 |
Correct |
17 ms |
24964 KB |
Output is correct |
16 |
Correct |
19 ms |
24780 KB |
Output is correct |
17 |
Correct |
18 ms |
24968 KB |
Output is correct |
18 |
Correct |
16 ms |
24880 KB |
Output is correct |
19 |
Correct |
16 ms |
24912 KB |
Output is correct |
20 |
Correct |
16 ms |
24952 KB |
Output is correct |
21 |
Correct |
16 ms |
24460 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
15 ms |
24496 KB |
Output is correct |
2 |
Correct |
15 ms |
24524 KB |
Output is correct |
3 |
Correct |
14 ms |
24588 KB |
Output is correct |
4 |
Correct |
16 ms |
25112 KB |
Output is correct |
5 |
Correct |
20 ms |
25316 KB |
Output is correct |
6 |
Correct |
17 ms |
24856 KB |
Output is correct |
7 |
Correct |
16 ms |
25216 KB |
Output is correct |
8 |
Correct |
16 ms |
24976 KB |
Output is correct |
9 |
Correct |
16 ms |
25280 KB |
Output is correct |
10 |
Correct |
16 ms |
25156 KB |
Output is correct |
11 |
Correct |
16 ms |
25228 KB |
Output is correct |
12 |
Correct |
15 ms |
24880 KB |
Output is correct |
13 |
Correct |
16 ms |
24932 KB |
Output is correct |
14 |
Correct |
17 ms |
25264 KB |
Output is correct |
15 |
Correct |
17 ms |
24964 KB |
Output is correct |
16 |
Correct |
19 ms |
24780 KB |
Output is correct |
17 |
Correct |
18 ms |
24968 KB |
Output is correct |
18 |
Correct |
16 ms |
24880 KB |
Output is correct |
19 |
Correct |
16 ms |
24912 KB |
Output is correct |
20 |
Correct |
16 ms |
24952 KB |
Output is correct |
21 |
Correct |
16 ms |
24460 KB |
Output is correct |
22 |
Correct |
15 ms |
24524 KB |
Output is correct |
23 |
Correct |
22 ms |
29152 KB |
Output is correct |
24 |
Correct |
33 ms |
32164 KB |
Output is correct |
25 |
Correct |
21 ms |
28096 KB |
Output is correct |
26 |
Correct |
25 ms |
28720 KB |
Output is correct |
27 |
Correct |
29 ms |
31268 KB |
Output is correct |
28 |
Correct |
24 ms |
28560 KB |
Output is correct |
29 |
Correct |
29 ms |
31508 KB |
Output is correct |
30 |
Incorrect |
29 ms |
31280 KB |
Output isn't correct |
31 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
572 ms |
218004 KB |
Output is correct |
2 |
Correct |
595 ms |
214636 KB |
Output is correct |
3 |
Correct |
481 ms |
226104 KB |
Output is correct |
4 |
Correct |
593 ms |
224712 KB |
Output is correct |
5 |
Correct |
494 ms |
226196 KB |
Output is correct |
6 |
Correct |
460 ms |
225116 KB |
Output is correct |
7 |
Correct |
547 ms |
225280 KB |
Output is correct |
8 |
Correct |
377 ms |
210088 KB |
Output is correct |
9 |
Correct |
511 ms |
211840 KB |
Output is correct |
10 |
Correct |
621 ms |
219540 KB |
Output is correct |
11 |
Correct |
487 ms |
225832 KB |
Output is correct |
12 |
Correct |
440 ms |
225548 KB |
Output is correct |
13 |
Correct |
15 ms |
24496 KB |
Output is correct |
14 |
Correct |
353 ms |
216044 KB |
Output is correct |
15 |
Correct |
380 ms |
237084 KB |
Output is correct |
16 |
Correct |
22 ms |
28508 KB |
Output is correct |
17 |
Correct |
22 ms |
28592 KB |
Output is correct |
18 |
Correct |
21 ms |
28592 KB |
Output is correct |
19 |
Correct |
22 ms |
28592 KB |
Output is correct |
20 |
Correct |
22 ms |
28584 KB |
Output is correct |
21 |
Correct |
371 ms |
206272 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
572 ms |
218004 KB |
Output is correct |
2 |
Correct |
595 ms |
214636 KB |
Output is correct |
3 |
Correct |
481 ms |
226104 KB |
Output is correct |
4 |
Correct |
593 ms |
224712 KB |
Output is correct |
5 |
Correct |
494 ms |
226196 KB |
Output is correct |
6 |
Correct |
460 ms |
225116 KB |
Output is correct |
7 |
Correct |
547 ms |
225280 KB |
Output is correct |
8 |
Correct |
377 ms |
210088 KB |
Output is correct |
9 |
Correct |
511 ms |
211840 KB |
Output is correct |
10 |
Correct |
621 ms |
219540 KB |
Output is correct |
11 |
Correct |
487 ms |
225832 KB |
Output is correct |
12 |
Correct |
440 ms |
225548 KB |
Output is correct |
13 |
Correct |
15 ms |
24496 KB |
Output is correct |
14 |
Correct |
353 ms |
216044 KB |
Output is correct |
15 |
Correct |
380 ms |
237084 KB |
Output is correct |
16 |
Correct |
22 ms |
28508 KB |
Output is correct |
17 |
Correct |
22 ms |
28592 KB |
Output is correct |
18 |
Correct |
21 ms |
28592 KB |
Output is correct |
19 |
Correct |
22 ms |
28592 KB |
Output is correct |
20 |
Correct |
22 ms |
28584 KB |
Output is correct |
21 |
Correct |
371 ms |
206272 KB |
Output is correct |
22 |
Correct |
15 ms |
24480 KB |
Output is correct |
23 |
Correct |
14 ms |
24496 KB |
Output is correct |
24 |
Correct |
15 ms |
24532 KB |
Output is correct |
25 |
Incorrect |
521 ms |
207996 KB |
Output isn't correct |
26 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
16 ms |
24496 KB |
Output is correct |
2 |
Correct |
29 ms |
30640 KB |
Output is correct |
3 |
Correct |
28 ms |
31024 KB |
Output is correct |
4 |
Correct |
22 ms |
28464 KB |
Output is correct |
5 |
Correct |
16 ms |
25008 KB |
Output is correct |
6 |
Correct |
17 ms |
25264 KB |
Output is correct |
7 |
Correct |
16 ms |
24960 KB |
Output is correct |
8 |
Correct |
23 ms |
27976 KB |
Output is correct |
9 |
Correct |
22 ms |
28592 KB |
Output is correct |
10 |
Correct |
20 ms |
24860 KB |
Output is correct |
11 |
Correct |
20 ms |
24984 KB |
Output is correct |
12 |
Correct |
241 ms |
145384 KB |
Output is correct |
13 |
Correct |
406 ms |
225536 KB |
Output is correct |
14 |
Correct |
650 ms |
280352 KB |
Output is correct |
15 |
Correct |
756 ms |
313712 KB |
Output is correct |
16 |
Correct |
798 ms |
313948 KB |
Output is correct |
17 |
Correct |
434 ms |
220120 KB |
Output is correct |
18 |
Correct |
15 ms |
24496 KB |
Output is correct |
19 |
Correct |
358 ms |
206208 KB |
Output is correct |
20 |
Correct |
16 ms |
24496 KB |
Output is correct |
21 |
Correct |
548 ms |
345652 KB |
Output is correct |
22 |
Correct |
642 ms |
392296 KB |
Output is correct |
23 |
Correct |
680 ms |
415620 KB |
Output is correct |
24 |
Correct |
407 ms |
230784 KB |
Output is correct |
25 |
Correct |
28 ms |
31228 KB |
Output is correct |
26 |
Correct |
34 ms |
32296 KB |
Output is correct |
27 |
Correct |
20 ms |
28204 KB |
Output is correct |
28 |
Correct |
16 ms |
25176 KB |
Output is correct |
29 |
Correct |
18 ms |
25212 KB |
Output is correct |
30 |
Correct |
17 ms |
24988 KB |
Output is correct |
31 |
Correct |
224 ms |
161068 KB |
Output is correct |
32 |
Correct |
424 ms |
253764 KB |
Output is correct |
33 |
Correct |
15 ms |
24496 KB |
Output is correct |
34 |
Correct |
348 ms |
215940 KB |
Output is correct |
35 |
Correct |
418 ms |
237056 KB |
Output is correct |
36 |
Correct |
15 ms |
24496 KB |
Output is correct |
37 |
Correct |
15 ms |
24524 KB |
Output is correct |
38 |
Correct |
14 ms |
24588 KB |
Output is correct |
39 |
Correct |
16 ms |
25112 KB |
Output is correct |
40 |
Correct |
20 ms |
25316 KB |
Output is correct |
41 |
Correct |
17 ms |
24856 KB |
Output is correct |
42 |
Correct |
16 ms |
25216 KB |
Output is correct |
43 |
Correct |
16 ms |
24976 KB |
Output is correct |
44 |
Correct |
16 ms |
25280 KB |
Output is correct |
45 |
Correct |
16 ms |
25156 KB |
Output is correct |
46 |
Correct |
16 ms |
25228 KB |
Output is correct |
47 |
Correct |
15 ms |
24880 KB |
Output is correct |
48 |
Correct |
16 ms |
24932 KB |
Output is correct |
49 |
Correct |
17 ms |
25264 KB |
Output is correct |
50 |
Correct |
17 ms |
24964 KB |
Output is correct |
51 |
Correct |
19 ms |
24780 KB |
Output is correct |
52 |
Correct |
18 ms |
24968 KB |
Output is correct |
53 |
Correct |
16 ms |
24880 KB |
Output is correct |
54 |
Correct |
16 ms |
24912 KB |
Output is correct |
55 |
Correct |
16 ms |
24952 KB |
Output is correct |
56 |
Correct |
16 ms |
24460 KB |
Output is correct |
57 |
Correct |
15 ms |
24524 KB |
Output is correct |
58 |
Correct |
22 ms |
29152 KB |
Output is correct |
59 |
Correct |
33 ms |
32164 KB |
Output is correct |
60 |
Correct |
21 ms |
28096 KB |
Output is correct |
61 |
Correct |
25 ms |
28720 KB |
Output is correct |
62 |
Correct |
29 ms |
31268 KB |
Output is correct |
63 |
Correct |
24 ms |
28560 KB |
Output is correct |
64 |
Correct |
29 ms |
31508 KB |
Output is correct |
65 |
Incorrect |
29 ms |
31280 KB |
Output isn't correct |
66 |
Halted |
0 ms |
0 KB |
- |