# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
897470 | mihtriii295 | Sjekira (COCI20_sjekira) | C++17 | 39 ms | 5844 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#pragma GCC optimize("O2")
#define ll long long
#define TIME (1.0 * clock() / CLOCKS_PER_SEC)
#define el cout << '\n'
using namespace std;
const ll N = 1e5 + 1;
const ll logN = 20;
const ll MOD = 1e9 + 7;
ll a[N], res, n;
struct edge{
int u, v;
};
vector<edge> adj;
bool cmp(edge x, edge y){
return max(a[x.u], a[x.v]) < max(a[y.u], a[y.v]);
}
struct DSU{
ll par[N], sz[N];
void init(ll n){
for (ll i = 1; i <= n; ++i){
sz[i] = 1;
par[i] = i;
}
}
ll Find(ll u){
if (u == par[u]) return u;
return par[u] = Find(par[u]);
}
ll Union(ll u, ll v){
u = Find(u);
v = Find(v);
if (u == v) return 0;
ll res = a[u] + a[v];
if (sz[u] < sz[v]) swap(u, v);
par[v] = u;
sz[u] += sz[v];
a[u] = max(a[u], a[v]);
a[v] = max(a[u], a[v]);
return res;
}
} dsu;
int main(){
if(fopen("coci2021_r2_sjekira.inp", "r")){
freopen("coci2021_r2_sjekira.inp", "r", stdin);
freopen("coci2021_r2_sjekira.out", "w", stdout);
}
ios_base::sync_with_stdio(0); cin.tie(NULL);
cout.tie(NULL);
cin >> n;
for (int i = 1; i <= n; ++i)
cin >> a[i];
for (int i = 1; i < n; ++i){
int u, v;
cin >> u >> v;
adj.push_back({u, v});
}
dsu.init(n);
sort(begin(adj), end(adj), cmp);
for (edge x : adj){
res += dsu.Union(x.u, x.v);
}
cout << res;
return 0;
}
컴파일 시 표준 에러 (stderr) 메시지
# | 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... |