poj - 1751 最小生成树kruskal 优化。

最新推荐文章于 2024-08-26 14:09:58 发布

原创最新推荐文章于 2024-08-26 14:09:58 发布 · 248 阅读

0 ·

CC 4.0 BY-SA版权

最小生成树专栏收录该内容

8 篇文章

订阅专栏

本文提供了一种解决POJ1751问题的有效算法，该问题涉及计算连接所有城市的最小生成树的代价，同时考虑已有的免费道路。通过使用Kruskal算法，文章详细介绍了如何构建和优化最小生成树，并附带了完整的C++代码实现。

题意：给你N个城市的坐标，城市之间存在公路，但是由于其中一些道路损坏了，需要维修，维修的费用与公路长成正比（公路是直的）。但现有M条公路是完整的，不需要维修，下面有M行，表示不需要维修的道路两端的城市，问最短费用。

直接做？哇无限wa。。。。。。

最后也不知道为啥，改了改数组大小啊，输入输出啊，什么的玄学优化就a了。。。。。。

然后一样的代码在c++会t。。。。。。

链接：poj 1751

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <limits>
#include <vector>
#include <sstream>
#define ll long long

using namespace std;

const int maxn = 2200 + 10;
const int maxm = 1001000;
const int inf = 0x3f3f3f3f;

int fa[maxn];
int rnk[maxn];

int num[maxn];

int n, m;

struct node {
    int a, b;
    int c;
    friend bool operator < (node x, node y) {
        return x.c < y.c;
    }
}q[maxm];

void init() {
    for(int i = 0; i <= n; i++) {
        fa[i] = i;
        rnk[i] = 0;
    }
}

int getf(int x) {
    if(x != fa[x]) {
        fa[x] = getf(fa[x]);
    }
    return fa[x];
}

void unions(int x, int y) {
    x = getf(x);
    y = getf(y);
    if(x == y) return;
    if(rnk[x] < rnk[y]) {
        fa[x] = y;
    }
    else {
        fa[y] = x;
        if(rnk[x] == rnk[y]) rnk[x]++;
    }
}

bool same(int x, int y) {
    return getf(x) == getf(y);
}

void kruskal() {
    sort(q + 1, q + 1 + m);
    int ans = 0;
    int nedge = 0;
    for(int i = 1; i <= m && nedge != n - 1; i++) {
        if(getf(q[i].a) != getf(q[i].b)) {
            unions(q[i].a, q[i].b);
            if(q[i].c != 0) {
                printf("%d %d\n", q[i].a, q[i].b);
            }
            nedge++;
        }
    }
    //if(nedge < n - 1) ans = -1;
    //return ans;
}

int x[maxn], y[maxn];

int dis(int i, int j) {
    return (x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]);
}

int main()
{
    int T, kcase = 0;
    while(~scanf("%d", &n)) {
        m = 0;
        for(int i = 1; i <= n; i++) {
            scanf("%d %d", &x[i], &y[i]);
            for(int j = 1; j < i; j++) {
                int res = dis(i, j);
                q[++m].a = i;
                q[m].b = j;
                q[m].c = res;
            }
        }
        int k;
        scanf("%d", &k);
        int a, b;
        init();
        for(int i = 1; i <= k; i++) {
            scanf("%d %d", &a, &b);
            unions(a, b);
        }
        kruskal();
    }
    return 0;
}