Submission #886228

#TimeUsernameProblemLanguageResultExecution timeMemory
886228javotazFactories (JOI14_factories)C++17
Compilation error
0 ms0 KiB
// In the Name of Allah #include<bits/stdc++.h> using namespace std; #define int long long #define F first #define S second #define pii pair<int, int> #define pb push_back #define pp pop_back #define all(x) x.begin(), x.end() #define cln(x) sort(all(x)), x.resize(unique(all(x)) - x.begin()) #define unq(a, n); {vector<int> v; for(int i = 0; i < n; i++) v.pb(a[i]); cln(v); for (int i = 0; i < n; i++) a[i] = lower_bound(all(v), a[i]) - v.begin();} const int N = 5e5 + 12, inf = 1e18 + 12; vector<pii> g[N]; int sz[N], val[N], cpar[N], q, n; set<pii> cdis[N]; bool mrk[N]; void dfs(int u, int par = -1) { sz[u] = 1; for (auto i: g[u]) if (i.F != par && !mrk[i.F]) { dfs(i.F, u); sz[u] += sz[i.F]; } } int find_centroid(int u, int c, int par = -1) { for (auto i: g[u]) if (!mrk[i.F] && i.F != par && sz[i.F] > (c - 1) / 2) return find_centroid(i.F, c, u); return u; } void fill_cdis(int u, int cent, int c = 0, int par = -1) { cdis[cent].insert({u, c}); for (auto i: g[u]) if (!mrk[i.F] && i.F != par) fill_cdis(i.F, cent, c + i.S, u); } void decompose(int u, int par = -1) { dfs(u); u = find_centroid(u, sz[u]); fill_cdis(u, u); cpar[u] = par; mrk[u] = true; for (auto i: g[u]) if (!mrk[i.F]) decompose(i.F, u); } void add(int u, int v) { if (u == -1) return; val[u] = min(val[u], cdis[u].lower_bound({v, -1})->S); add(cpar[u], v); } int ask(int u, int v) { if (u == -1) return inf; int tmp = cdis[u].lower_bound({v, -1})->S; return min(ask(cpar[u], v), tmp + val[u]); } void rem(int u) { val[u] = inf; if (cpar[u] != -1) rem(cpar[u]); } void Init(int c, int a[], int b[], int d[]) { n = c; for (int i = 1; i < n; i++) { g[a[i]].pb({b[i], d[i]}); g[b[i]].pb({a[i], d[i]}); } decompose(0); fill(val, val + n, inf); } int Query(int s, int x[], int t, int y[]) { for (int i = 0; i < s; i++) add(x[i], x[i]); int res = inf; for (int i = 0; i < t; i++) res = min(res, ask(y[i], y[i])); for (int i = 0; i < s; i++) rem(x[i]); return res; }

Compilation message (stderr)

/usr/bin/ld: /tmp/cckjT4u6.o: in function `main':
grader.cpp:(.text.startup+0x37d): undefined reference to `Init(int, int*, int*, int*)'
/usr/bin/ld: grader.cpp:(.text.startup+0x412): undefined reference to `Query(int, int*, int, int*)'
collect2: error: ld returned 1 exit status