BOJ 16562 - 친구비

https://www.acmicpc.net/problem/16562

16562번: 친구비

유니온파인드 문제다. 각 집합의 최소를 모두 더한 합을 출력하면 된다. 

#include <iostream>
#include <algorithm>
using namespace std;

int n, m, k;
int uf[10001];
int cost[10001];
int rst[10001];

int find(int a) {
	if (uf[a] == a) return a;
	return uf[a] = find(uf[a]);
}

bool merge(int a, int b) {
	a = find(a);
	b = find(b);
	if (a == b) return false;
	uf[a] = b;
	return true;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cin >> n >> m >> k;
	for (int i = 0; i < n; i++) {
		cin >> cost[i];
		uf[i] = i;
	}

	for (int i = 0,u,v; i < m; i++) {
		cin >> u >> v;
		merge(u - 1, v - 1);
	}

	fill(rst, rst + n, 1e9);
	for (int i = 0; i < n; i++) rst[find(i)] = min(rst[find(i)],cost[i]);
	
	int sum = 0;
	for (int i = 0; i < n; i++) {
		if (rst[i] >= 1e9) continue;
		sum += rst[i];
	}
	if (sum <= k) cout << sum << '\n';
	else cout << "Oh no\n";
}

왜 리뷰하라고 메모해뒀지..?