이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
#define nmax 1000100
int n, sum, a[nmax], s[nmax], d[nmax];
vector <int> adj[nmax];
int val[nmax], res[nmax];
void dfs1(int u, int parent)
{
res[u] = val[u] = 0;
for(int v: adj[u]) {
if(v != parent) {
dfs1(v, u);
val[u] += a[v] + val[v];
res[u] = max(res[u], a[v] + val[v]);
}
}
res[u] = max(res[u], sum - a[u] - val[u]);
}
int LocateCentre(int n, int a[], int s[], int d[])
{
sum = 0;
for(int i = 0; i < n; ++i)
sum += a[i];
for(int i = 0; i < n - 1; ++i) {
adj[s[i]].push_back(d[i]);
adj[d[i]].push_back(s[i]);
}
dfs1(0, 0);
return min_element(res, res + n) - res;
}
/* int main()
* {
* ios::sync_with_stdio(false); cin.tie(NULL);
*
* cin >> n;
* for(int i = 0; i < n; ++i)
* cin >> a[i];
* for(int i = 0; i < n - 1; ++i)
* cin >> s[i];
* for(int i = 0; i < n - 1; ++i)
* cin >> d[i];
* cout << LocateCentre(n, a, s, d) << '\n';
*
* return 0;
* }
*/
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |