16-yuyu0830

yuyu0830/README.md

@@ -9,5 +9,6 @@
 | 5차시 | 2024.03.29 |  탐색  | [스도쿠](https://www.acmicpc.net/problem/2239)  | - |
 | 6차시 | 2024.04.03 |  자료구조  | [나만 안되는 연애](https://www.acmicpc.net/problem/14621)  | - |
 | 7차시 | 2024.04.04 |  저울  | [저울](https://www.acmicpc.net/problem/2437)  | - |
-| 10차시| 2024.04.29 |  백트래킹  | [비숍](https://www.acmicpc.net/problem/1799)  | - |
+| 10차시 | 2024.04.29 |  백트래킹  | [비숍](https://www.acmicpc.net/problem/1799)  | - |
+| 16차시 | 2024.06.04 |  별자리 만들기  | [별자리 만들기](https://www.acmicpc.net/problem/4386)  | - |
 ---

yuyu0830/자료구조/4386.cpp

@@ -0,0 +1,83 @@
+#include <iostream>
+#include <queue>
+#include <math.h>
+
+#define fastio cin.tie(NULL); cin.sync_with_stdio(false);
+
+using namespace std;
+
+struct point {
+    float x, y;
+    point(float X, float Y) : x(X), y(Y) {};
+    point() {};
+};
+
+point arr[10000];
+
+struct edge {
+    int vertex1, vertex2;
+    float value;
+    edge(int v1, int v2) : vertex1(v1), vertex2(v2) {
+        value = sqrt(pow(arr[v1].x - arr[v2].x, 2) + pow(arr[v1].y - arr[v2].y, 2));
+    }
+};
+
+struct cmp {
+    bool operator()(edge a, edge b) { return a.value > b.value; }
+};
+
+int parent[101] = {0, };
+
+int find(int v) {
+    if (parent[v] == v) return v;
+    return parent[v] = find(parent[v]);
+}
+
+void merge(int x, int y) { parent[find(x)] = find(y); }
+
+bool isUnion(int x, int y) { return find(x) == find(y); }
+
+int main() {
+    fastio
+    int n; cin >> n;
+
+    for (int i = 1; i <= n; i++)
+        parent[i] = i;
+
+    priority_queue<edge, vector<edge>, cmp> q;
+
+    // 별 위치 저장
+    for (int i = 0; i < n; i++) {
+        float a, b; cin >> a >> b;
+        arr[i] = point(a, b);
+    }
+
+    // 각 별마다 이을 수 있는 모든 선 우선 순위 큐에 저장
+    // 우선 순위 큐는 선분의 길이를 기준으로 정렬
+    for (int i = 0; i < n; i++) 
+        for (int j = i + 1; j < n; j++) 
+            q.push(edge(i, j));
+
+    int cnt = 1;
+    float ans = 0.f;
+
+    while (cnt != n) {
+        edge tmp = q.top();
+        q.pop();
+
+        int v1 = tmp.vertex1;
+        int v2 = tmp.vertex2;
+
+        if (isUnion(v1, v2)) continue;
+
+        merge(v1, v2);
+        cnt++;
+        ans += tmp.value;
+
+        if (cnt == n) {
+            printf("%.2f\n", ans);
+            return 0;
+        }
+    }
+
+}