This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "traffic.h"
#include <bits/stdc++.h>
using namespace std;
int LocateCentre(int N, int pp[], int S[], int D[])
{
vector<set<int>> adj(N + 1);
for (int i = 0; i < N - 1; i++)
{
adj[S[i]].insert(D[i]);
adj[D[i]].insert(S[i]);
}
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> st;
for (int i = 0; i < N; i++)
if (adj[i].size() == 1)
st.push({pp[i], i});
while(st.size() >= 2)
{
int c = st.top().second;
int p = st.top().first;
st.pop();
auto nxt = *adj[c].begin();
adj[nxt].erase(c);
pp[nxt] += p;
if(adj[nxt].size() == 1)
st.push({pp[nxt], nxt});
}
return st.top().second;
}
# | 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... |