취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1197번 문제인 최소 스패닝 트리이다.
문제는 아래 링크를 확인하자.

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

이 문제는 주어진 그래프에서 MST(Minimum Spanning Tree: 최소신장트리)의 간선의 가중치의 합을 구하는 문제이다.

글쓴이는 Kruskal(크루스칼) 알고리즘을 이용하려 이 문제를 해결하였다.

아래의 구현에서, 배열 arr은 각 꼭짓점의 disjoint set(자료구조)을 나타내고, findf는 disjoint set에서 각 꼭짓점을 대표하는 꼭짓점이 무엇인지를 돌려주는 함수이다.

에지를 가중치순으로 정렬하고, 그 에지가 잇는 양 끝점이 연결되어있는가의 여부를 disjoint set을 통해 확인한 다음 연결되어있지 않다면 이 간선을 이용해 연결하는 것으로 Kruskal 알고리즘을 구현할 수 있다.

아래는 제출한 소스코드이다.

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

int arr[10001];

vector<pair<int, pair<int, int>>> edges;

int findf(int x) {
	if (x == arr[x]) return x;
	else return arr[x] = findf(arr[x]);
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int V, E; cin >> V >> E;
	for (int i = 1; i <= V; i++) {
		arr[i] = i;
	}

	while (E--) {
		int a, b, c; cin >> a >> b >> c;
		edges.push_back({ c,{a,b} });
	}

	sort(edges.begin(), edges.end());

	int ans = 0;
	for (auto edge : edges) {
		int x = findf(edge.second.first), y = findf(edge.second.second);
		if (x == y) continue;
		ans += edge.first;
		arr[y] = x;
	}

	cout << ans;
}