티스토리 뷰

백준

백준 소스코드 [C++] 1922 네트워크 연결

Hani_Levenshtein 2020. 8. 30. 17:24

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

 

1922번: 네트워크 연결

이 경우에 1-3, 2-3, 3-4, 4-5, 4-6을 연결하면 주어진 output이 나오게 된다.

www.acmicpc.net

백준 소스코드 [C++] 1922 네트워크 연결

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
#include <limits.h>
typedef long long ll;
using namespace std;

int Vertex, Edge, v, e, w;
vector<pair<int,pair<int, int> > >E;

struct DisjointSet {
	vector <int> Parent, Rank;
	DisjointSet(int n) :Parent(n), Rank(n, 1) {
		for (int i = 0;i < n;i++)Parent[i] = i;
	}
	int Find_Parent(int u)
	{
		if (Parent[u] == u) return u;
		else return Parent[u] = Find_Parent(Parent[u]);
	}
	void Merge(int u, int v)
	{
		u = Find_Parent(u);v = Find_Parent(v);
		if (u == v)return;
		if (Rank[u] > Rank[v]) swap(u, v);
		Parent[u] = v;
		if (Rank[u] == Rank[v]) Rank[v]++;
	}

};

int kruskal() {
	for (int i = 0;i < Edge;i++) {
		cin >> v >> e >> w;
		if (v != e)E.push_back({ w,{ v - 1,e - 1 } });
	}
	sort(E.begin(), E.end());
	DisjointSet sets(Vertex);
	int res = 0;
	for (int i = 0;i < (int)E.size();i++) {
		int cost = E[i].first;
		int from = E[i].second.first;
		int to = E[i].second.second;
		if (sets.Find_Parent(from) == sets.Find_Parent(to))continue;
		else {
			sets.Merge(from, to);
			res = res + cost;
		}
	}
	return res;
}
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cin >> Vertex;
	cin >> Edge;
	cout << kruskal();
	return 0;
}
댓글