#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define si second
typedef pair<int,int> pi;
#ifdef LOCAL
#define debug(...) __f(#__VA_ARGS__, __VA_ARGS__)
#else
#define debug(...) 69
#endif
template <typename Arg>
void __f(string name, Arg arg) {
cerr << name << " = " << arg << endl;
}
template <typename Head, typename... Tail>
void __f(string names, Head head, Tail... tail) {
string cur = "";
for (auto ch: names){if(ch==','){break;}else{cur+=ch;}}
string nxt = names.substr(cur.size()+2);
cerr << cur << " = " << head << ", ";
__f(nxt, tail...);
}
const int MXN = 1000005, INF = 1e9;
int sz[MXN], ans[MXN], ret = INF, total, num[MXN];
vector<int> adj[MXN];
void dfs(int x, int p) {
for (auto i: adj[x]) if (i != p) {
dfs(i, x);
num[x] += num[i] + sz[i];
ans[x] = max(ans[x], num[x]);
}
ans[x] = max(ans[x], total - num[x] - sz[x]);
}
int LocateCentre(int N, int P[], int S[], int D[]) {
for (int i = 0; i < N; ++i) {
total += P[i];
sz[i] = P[i];
}
for (int i = 0; i < N - 1; ++i) {
adj[S[i]].pb(D[i]);
adj[D[i]].pb(S[i]);
}
dfs(0, 0);
for (int i = 0; i < N; ++i) {
ret = min(ret, ans[i]);
// debug(ans[i], i);
}
return ret;
}
Compilation message
/tmp/cck2qowX.o: In function `main':
grader.cpp:(.text.startup+0xd9): undefined reference to `LocateCentre(int, int*, int*, int*)'
collect2: error: ld returned 1 exit status
