문제
https://www.acmicpc.net/problem/14621
문제해설
경로의 길이를 구하는 문제이다.
다만 조건이 있는데
도로는 남초 대학교와 여초 대학교만을 이어야하고
최단 경로로 이루어져야 한다.
문제풀이
결국 이 문제는 MST를 만드는 문제인데
경로를 만들 때 두 노드가 남초와 여초인지만 판단해주면 된다.
MST가 이루어지지 않는다면 -1 출력
전체코드
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
#define MAX 100001
using namespace std;
int parent[MAX];
char flag[1001]; //남초인지 여초인지 저장
struct edge {
int st, end, distance;
};
struct cmp {
bool operator()(edge x, edge y) {
return x.distance > y.distance;
}
};
int find(int x) {
if (parent[x] == x)
return x;
return parent[x] = find(parent[x]);
}
void merge(int x, int y) {
x = find(x);
y = find(y);
if (x == y)
return;
parent[y] = x;
}
bool isUnion(int x, int y) {
x = find(x);
y = find(y);
if (x == y)
return true;
return false;
}
int main() {
ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
priority_queue<edge, vector<edge>, cmp>arr;
int n, m; cin >> n >> m;
for (int i = 1; i <= n; i++)
cin >> flag[i];
for (int i = 0; i < m; i++) {
int a, b, c; cin >> a >> b >> c;
arr.push({ a,b,c });
}
for (int i = 0; i < MAX; i++)
parent[i] = i;
int sum = 0;
while (!arr.empty()) {
if (!isUnion(arr.top().st, arr.top().end) &&
flag[arr.top().st] != flag[arr.top().end]) {
sum += arr.top().distance;
merge(arr.top().st, arr.top().end);
}
arr.pop();
}
for (int i = 2; i <= n; i++) {
if (!isUnion(1, i)) {
cout << -1;
return 0;
}
}
cout << sum;
return 0;
}