#include "factories.h"
#include <bits/stdc++.h>
using namespace std;
#define FOR(i, j, k, l) for(int i = (j); i < (k); i += (l))
#define FORD(i, j, k, l) for(int i = (j); i >= (k); i -= (l))
#define REP(i, n) FOR(i, 0, n, 1)
#define REPD(i, n) FORD(i, n, 0, 1)
#define pb push_back
const int INF = (int)1e9;
typedef long long ll;
const ll INFF = (ll)1e18;
int n;
vector<vector<array<ll, 2>>> vg;
vector<vector<array<ll, 2>>> g;
vector<ll> depth;
vector<int> lev;
vector<vector<int>> nxt;
vector<int> in;
vector<int> out;
vector<bool> type_a;
vector<bool> type_b;
vector<array<ll, 2>> dp;
int t = 0;
void dfs(int v, int p, ll d){
in[v] = t++;
depth[v] = d;
if(p != -1){
lev[v] = lev[p] + 1;
}
nxt[0][v] = p;
FOR(i, 1, 20, 1){
if(nxt[i - 1][v] != -1){
nxt[i][v] = nxt[i - 1][nxt[i - 1][v]];
}
}
for(auto x : g[v]){
if(x[0] == p) continue;
dfs(x[0], v, d + x[1]);
}
out[v] = t;
}
int lca(int u, int v){
if(lev[u] < lev[v])
swap(u, v);
FORD(i, 19, 0, 1){
if(lev[u] - (1 << i) >= lev[v]){
u = nxt[i][u];
}
}
if(u == v)
return u;
REPD(i, 19){
if(nxt[i][u] != nxt[i][v]){
u = nxt[i][u];
v = nxt[i][v];
}
}
return nxt[0][u];
}
int build_virtual_tree(vector<int> &vertices){
auto upper = [&](int u, int v){
return in[u] <= in[v] && out[u] >= out[v];
};
auto cmp = [&](int u, int v){
return in[u] < in[v];
};
sort(vertices.begin(), vertices.end(), cmp);
int up = vertices.size();
REP(i, up - 1){
vertices.pb(lca(vertices[i], vertices[i + 1]));
}
sort(vertices.begin(), vertices.end(), cmp);
vertices.erase(unique(vertices.begin(), vertices.end()), vertices.end());
for(int v : vertices){
vg[v].clear();
type_a[v] = false;
type_b[v] = false;
}
vector<int> stk;
stk.pb(vertices[0]);
FOR(i, 1, vertices.size(), 1){
while(stk.size() >= 2 && !upper(stk.back(), vertices[i])){
int u = stk.back();
stk.pop_back();
int v = stk.back();
ll w = abs(depth[u] - depth[v]);
vg[u].pb({v, w});
vg[v].pb({u, w});
}
stk.pb(vertices[i]);
}
while(stk.size() >= 2){
int u = stk.back();
stk.pop_back();
int v = stk.back();
ll w = abs(depth[u] - depth[v]);
vg[u].pb({v, w});
vg[v].pb({u, w});
}
return stk[0];
}
void solve(int v, int p, ll &ans){
dp[v] = {INFF, INFF};
for(auto x : vg[v]){
if(x[0] == p) continue;
solve(x[0], v, ans);
auto ar = dp[x[0]];
ar[0] += x[1];
ar[1] += x[1];
dp[v][0] = min(dp[v][0], ar[0]);
dp[v][1] = min(dp[v][1], ar[1]);
}
if(type_a[v])
dp[v][0] = 0;
if(type_b[v])
dp[v][1] = 0;
ans = min(ans, dp[v][0] + dp[v][1]);
}
void Init(int N, int A[], int B[], int D[]) {
n = N;
vg.resize(n);
g.resize(n);
depth.resize(n);
nxt = vector<vector<int>>(20, vector<int>(n, -1));
in.resize(n);
out.resize(n);
lev.resize(n, 0);
type_a.resize(n, false);
type_b.resize(n, false);
dp.resize(n);
REP(i, n - 1){
g[A[i]].pb({B[i], D[i]});
g[B[i]].pb({A[i], D[i]});
}
dfs(0, -1, 0);
}
long long Query(int S, int X[], int T, int Y[]) {
vector<int> vertices;
REP(i, S){
vertices.pb(X[i]);
}
REP(i, T){
vertices.pb(Y[i]);
}
int root = build_virtual_tree(vertices);
REP(i, S){
type_a[X[i]] = true;
}
REP(i, T){
type_b[Y[i]] = true;
}
ll ans = INFF;
solve(root, -1, ans);
return ans;
}
Compilation message
factories.cpp: In function 'int build_virtual_tree(std::vector<int>&)':
factories.cpp:5:44: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
5 | #define FOR(i, j, k, l) for(int i = (j); i < (k); i += (l))
| ^
factories.cpp:82:5: note: in expansion of macro 'FOR'
82 | FOR(i, 1, vertices.size(), 1){
| ^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
24 ms |
716 KB |
Output is correct |
2 |
Correct |
1014 ms |
9808 KB |
Output is correct |
3 |
Correct |
1076 ms |
9692 KB |
Output is correct |
4 |
Correct |
1026 ms |
9800 KB |
Output is correct |
5 |
Correct |
892 ms |
10092 KB |
Output is correct |
6 |
Correct |
693 ms |
9668 KB |
Output is correct |
7 |
Correct |
1059 ms |
9764 KB |
Output is correct |
8 |
Correct |
1024 ms |
9864 KB |
Output is correct |
9 |
Correct |
867 ms |
9924 KB |
Output is correct |
10 |
Correct |
696 ms |
9612 KB |
Output is correct |
11 |
Correct |
1094 ms |
9748 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
3 ms |
460 KB |
Output is correct |
2 |
Correct |
2679 ms |
133320 KB |
Output is correct |
3 |
Correct |
3677 ms |
136536 KB |
Output is correct |
4 |
Correct |
1848 ms |
130704 KB |
Output is correct |
5 |
Correct |
3429 ms |
165780 KB |
Output is correct |
6 |
Correct |
3952 ms |
138344 KB |
Output is correct |
7 |
Correct |
3375 ms |
36460 KB |
Output is correct |
8 |
Correct |
1877 ms |
35704 KB |
Output is correct |
9 |
Correct |
2419 ms |
41468 KB |
Output is correct |
10 |
Correct |
3374 ms |
37664 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
24 ms |
716 KB |
Output is correct |
2 |
Correct |
1014 ms |
9808 KB |
Output is correct |
3 |
Correct |
1076 ms |
9692 KB |
Output is correct |
4 |
Correct |
1026 ms |
9800 KB |
Output is correct |
5 |
Correct |
892 ms |
10092 KB |
Output is correct |
6 |
Correct |
693 ms |
9668 KB |
Output is correct |
7 |
Correct |
1059 ms |
9764 KB |
Output is correct |
8 |
Correct |
1024 ms |
9864 KB |
Output is correct |
9 |
Correct |
867 ms |
9924 KB |
Output is correct |
10 |
Correct |
696 ms |
9612 KB |
Output is correct |
11 |
Correct |
1094 ms |
9748 KB |
Output is correct |
12 |
Correct |
3 ms |
460 KB |
Output is correct |
13 |
Correct |
2679 ms |
133320 KB |
Output is correct |
14 |
Correct |
3677 ms |
136536 KB |
Output is correct |
15 |
Correct |
1848 ms |
130704 KB |
Output is correct |
16 |
Correct |
3429 ms |
165780 KB |
Output is correct |
17 |
Correct |
3952 ms |
138344 KB |
Output is correct |
18 |
Correct |
3375 ms |
36460 KB |
Output is correct |
19 |
Correct |
1877 ms |
35704 KB |
Output is correct |
20 |
Correct |
2419 ms |
41468 KB |
Output is correct |
21 |
Correct |
3374 ms |
37664 KB |
Output is correct |
22 |
Correct |
4567 ms |
143748 KB |
Output is correct |
23 |
Correct |
4310 ms |
144184 KB |
Output is correct |
24 |
Correct |
4609 ms |
147844 KB |
Output is correct |
25 |
Correct |
4876 ms |
150560 KB |
Output is correct |
26 |
Correct |
5394 ms |
142344 KB |
Output is correct |
27 |
Correct |
4283 ms |
168548 KB |
Output is correct |
28 |
Correct |
3019 ms |
138908 KB |
Output is correct |
29 |
Correct |
5722 ms |
141116 KB |
Output is correct |
30 |
Correct |
5665 ms |
140480 KB |
Output is correct |
31 |
Correct |
5666 ms |
140992 KB |
Output is correct |
32 |
Correct |
1964 ms |
45112 KB |
Output is correct |
33 |
Correct |
1715 ms |
38488 KB |
Output is correct |
34 |
Correct |
2605 ms |
35552 KB |
Output is correct |
35 |
Correct |
2632 ms |
35260 KB |
Output is correct |
36 |
Correct |
2899 ms |
36232 KB |
Output is correct |
37 |
Correct |
3065 ms |
36188 KB |
Output is correct |