3.1. 15이원준 ¶
#include<iostream>
#include<vector>
using namespace std;
typedef struct node {
int s, e, v;
node(int a, int b, int c) {
s = a;
e = b;
v = c;
}
}node;
int visit[1010] = { 0, };
vector<node> in;
vector<int> checked;
int N, M;
bool choose(int e) {
if ((visit[in[e].e] == 1 && visit[in[e].s] == 1) || (visit[in[e].e] == 0 && visit[in[e].s] == 0)) {
return false;
}
return true;
}
int main() {
cin >> N >> M;
for (int i = 0; i<M; i++) {
int a, b, c;
cin >> a >> b >> c;
in.push_back(node(a, b, c));
}
for (int i = 0; i<N - 1; i++) {
int min = -1;
for (int j = 0; j<M; j++) {
if ((checked.size() == 0 || choose(j)) && (min == -1 || in[min].v > in[j].v)) {
min = j;
}
}
visit[in[min].s] = 1;
visit[in[min].e] = 1;
checked.push_back(min);
}
int answer = 0;
for (int i = 0; i<N - 1; i++) {
answer += in[checked[i]].v;
}
cout << answer << endl;
}
3.2. 박인서 ¶
#include <iostream>
#include <vector>
#include <algorithm>
typedef std::pair<int, int> pair_int;
std::vector<pair_int> g[10001];
bool visit[10001];
int main()
{
int n, m, res = 0;
std::cin >> n >> m;
for (int i = 0; i < m; i++) {
int a, b, c;
std::cin >> a >> b >> c;
g[a].push_back(pair_int(b, c));
g[b].push_back(pair_int(a, c));
}
visit[1] = true;
for (int i = 1; i < n; i++) {
int x, y, min = 217483647;
for (int j = 1; j <= n; j++) {
if (visit[j]) {
for (int k = 0; k < g[j].size(); k++) {
if (!visit[g[j][k].first] && g[j][k].second < min)
min = g[j][k].second, x = j, y = g[j][k].first;
}
}
}
visit[y] = true;
res += min;
}
std::cout << res;
return 0;
}
4.2. 박인서 ¶
- 프림 알고리즘을 이용하였습니다.
크루스칼로 구현하기 위해서는 Union Find를 구현해야 되는데 이해가 안가서..
- 자세한 사항은 프림 알고리즘-위키피디아 참조










