#include <bits/stdc++.h>
#define f first
#define s second
#define pb push_back
using namespace std;
typedef long long ll;
typedef long double ld;
const ll N = 1e6 + 100;
const ll oo = 1e10;
ll a[N];
ll pred[N];
ll r[N];
ll cmp(ll x)
{
if (x == pred[x]) return x;
else return (pred[x] = cmp(pred[x]));
}
void unite(ll x, ll y)
{
if (r[x] > r[y]) pred[y] = x;
else
{
pred[x] = y;
r[y] += (r[x] == r[y]);
}
a[y] = max(a[y], a[x]);
a[x] = max(a[x], a[y]);
}
int main()
{
ll n;
cin >> n;
for (ll i = 0; i < n; i++) cin >> a[i];
int mx[n];
for (int i = 0; i < n; i++) mx[i] = a[i];
for (ll i = 0; i < n; i++) pred[i] = i, r[i] = 0;
vector <pair<ll, ll> > rebra;
for (ll i = 0; i < n - 1; i++)
{
ll x, y;
cin >> x >> y;
rebra.pb({--x, --y});
}
bool mrk[n - 1];
for (ll i = 0; i < n - 1; i++) mrk[i] = 0;
ll ans = 0;
for (ll i = 0; i < n - 1; i++)
{
ll mx1 = oo, nom = -1;
for (ll j = 0; j < n - 1; j++)
if (!mrk[j])
{
ll x = rebra[j].f, y = rebra[j].s;
if (mx[cmp(x)] + mx[cmp(y)] < mx1) mx1 = mx[cmp(x)] + mx[cmp(y)], nom = j;
}
ans += a[cmp(rebra[nom].f)] + a[cmp(rebra[nom].s)];
mrk[nom] = 1;
unite(cmp(rebra[nom].f), cmp(rebra[nom].s));
}
cout << ans;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
1099 ms |
4704 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
364 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |