'commit'

TangDou/AcWing_TiGao/T3/MinialSpanningTree/1144.cpp

@@ -0,0 +1,59 @@
+#include <bits/stdc++.h>
+using namespace std;
+const int N = 1000 * 1000 + 10, M = N << 1;
+int n, m;
+
+// 二维转一维的办法,坐标从(1,1)开始
+int get(int x, int y) {
+    // m为列宽
+    return (x - 1) * m + y;
+}
+struct Edge {
+    int a, b, c;
+    const bool operator<(const Edge &t) const {
+        return c < t.c;
+    }
+} edge[M];
+int el;
+
+// 并查集
+int p[N];
+int find(int x) {
+    if (p[x] != x) p[x] = find(p[x]);
+    return p[x];
+}
+// 先连1的边，再连2的边
+void create_edges() {
+    for (int i = 1; i <= (n - 1) * m; i++) // 前n-1行的每个点，都可以向下引一条边权为1的边
+        edge[el++] = {i, i + m, 1};        // i -> i+m,边权为1
+
+    for (int i = 1; i <= n * m; i++) { // 向右引边，注意最后一列不能向右引边
+        if (i % m == 0) continue;      // 最后一列放过
+        edge[el++] = {i, i + 1, 2};    // i->i+1,边权为2
+    }
+    // 因为加进去就是按边权由小到大录入的，所以不用再排序了
+}
+int main() {
+    cin >> n >> m;
+    // 建边
+    create_edges();
+
+    // 并查集初始化
+    for (int i = 1; i <= n * m; i++) p[i] = i;
+
+    int x1, y1, x2, y2;
+    // 利用二维转一维办法，将(x,y)映射成节点编号
+    // 某些点之间已经有连线了=> 标识这些点已在并查集中
+    while (cin >> x1 >> y1 >> x2 >> y2) {
+        int a = get(x1, y1), b = get(x2, y2);
+        p[find(a)] = find(b);
+    }
+
+    int res = 0; // 用Kruskal算法即可
+    for (int i = 0; i < el; i++) {
+        int a = find(edge[i].a), b = find(edge[i].b), c = edge[i].c;
+        if (a != b) p[a] = b, res += c;
+    }
+    cout << res << endl;
+    return 0;
+}