Submission #365167

#	Time	Username	Problem	Language	Result	Execution time	Memory
365167		mohamedsobhi777	Sjekira (COCI20_sjekira)	C++14	110 / 110	121 ms	6380 KiB

sjekira

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>

#pragma GCC optimize("-Ofast")
//#pragma GCC optimize("trapv")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,sse4.2,popcnt,abm,mmx,avx2,tune=native")
#pragma GCC optimize("-ffast-math")
#pragma GCC optimize("-funroll-loops")

using namespace std;
using namespace __gnu_pbds;

#define vi vector<int>
#define vll vector<ll>
#define vii vector<pair<int, int>>
#define vvi vector<vi>
#define vvii vector<vii>
#define pii pair<int, int>
#define pll pair<ll, ll>
#define loop(_) for (int __ = 0; __ < (_); ++__)
#define forn(i, n) for (int i = 0; i < n; ++i)
#define pb push_back
#define f first
#define s second
#define sz(_) ((int)_.size())
#define all(_) _.begin(), _.end()
#define uni(_) unique(_)
#define lb lower_bound
#define ub upper_bound
#define si set<int>
#define ms multiset<int>
#define qi queue<int>
#define pq prioriry_queue<int>
#define mi map<int, int>
#define inc(i, l, r) for (int i = l; i <= r; i++)
#define dec(i, l, r) for (int i = l; i >= r; i--)

using lll = __int128;
using ll = long long;
using ld = long double;

const int N = 1e5 + 7;
const ll mod = 1e9 + 7;
const ll inf = 2e18;

auto ra = [] {char *p = new char ; delete p ; return ll(p) ; };
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count() * (ra() | 1));
typedef tree<pii, null_type, less<pii>, rb_tree_tag, tree_order_statistics_node_update> os;

int n;

struct dsu
{
       int fat[N];
       int mx[N];
       dsu()
       {
              iota(fat, fat + N, 0);
       }
       int find(int x) { return fat[x] = (x == fat[x] ? x : find(fat[x])); }
       void link(int u, int v)
       {
              u = find(u), v = find(v);
              mx[v] = max(mx[v], mx[u]);
              fat[u] = v;
       }
       bool same(int u, int v)
       {
              return find(u) == find(v);
       }
} de;
priority_queue<pair<int, pii>, vector<pair<int, pii>>, greater<pair<int, pii>>> pr;

int main()
{
       ios_base::sync_with_stdio(0);
       cin.tie(0);
#ifndef ONLINE_JUDGE
#endif
       cin >> n;
       for (int i = 1; i <= n; ++i)
              cin >> de.mx[i];

       vii ed;

       for (int i = 0; i < n - 1; ++i)
       {
              int u, v;
              cin >> u >> v;
              ed.pb({u, v});
              pr.push({de.mx[u] + de.mx[v], {u, v}});
       }
       ll ans = 0;
       while (sz(pr))
       {
              pair<int, pii> tp = pr.top();
              pr.pop();
              if (de.same(tp.s.f, tp.s.s))
              {
                     continue;
              }
              if (de.mx[de.find(tp.s.f)] + de.mx[de.find(tp.s.s)] != tp.f)
              {
                     pr.push({de.mx[de.find(tp.s.f)] + de.mx[de.find(tp.s.s)], tp.s});
                     continue;
              }
              ans += tp.f;
              de.link(tp.s.f, tp.s.s);
       }
       cout << ans;
       return 0;
}

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 30 / 30
• Subtask 4: 60 / 60