UVa11631 Dark Roads（最小生成树，邻接表）

最新推荐文章于 2024-08-03 15:53:55 发布

AcDora

最新推荐文章于 2024-08-03 15:53:55 发布

阅读量849

点赞数

CC 4.0 BY-SA版权

分类专栏：最小生成树、并查集

本文链接：https://blog.youkuaiyun.com/ac_lion/article/details/8583953

最小生成树、并查集专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种针对大规模数据集求解最小生成树的算法，利用邻接表进行存储优化，并通过Prim算法高效求解。实例展示了如何将Prim算法应用于实际问题中，以减少计算复杂度并提高处理大型数据集的效率。

很裸的一道求最小生成树的题，只是数据很大，要用邻接表来存储

代码：

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int N = 200010;
const long long INF = 10000000000;
int n, m, id, head[N];
long long sum;
struct edge{
    int t, next;
    int l;
}e[N*2];
long long prim() {
    int u = n - 1, v;
    long long mi, ans = 0, d[N];
    bool vis[N];
    for ( int i = 0; i < n; ++i ) d[i] = INF;
    memset(vis, 0, sizeof(vis));
    d[0] = 0;
    
    for ( int t = 0; t <= u; ++t ) {
        mi = INF;
        for ( int i = 0; i < n; ++i ) if ( !vis[i] && d[i] < mi ) {
            v = i, mi = d[i];
        }
            //printf("%d %lld\n", v, mi);
        if ( mi == INF ) break;
        ans += mi;
        vis[v] = true;
        for ( int i = head[v]; i != -1; i = e[i].next ) {
            int x = e[i].t, c = e[i].l;
            if ( !vis[x] && d[x] > c ) d[x] = c;
        }
    }
    return ans;
}
void add( int u, int v, int c ) {
    e[id].next = head[u], e[id].t = v, e[id].l = c, head[u] = id++;
    e[id].next = head[v], e[id].t = u, e[id].l = c, head[v] = id++;
}

int main()
{
    while ( scanf("%d%d", &n, &m) != EOF && !( m == 0 && n == 0 ) ) {
        sum = 0, id = 0;
        for ( int i = 0; i <= n; ++i ) head[i] = -1;
        while ( m-- ) {
            int s, t, l;
            scanf("%d%d%d", &s, &t, &l);
            sum += l;
            add( s, t, l );
        }
        printf("%lld\n", sum - prim());
    }
}