浙大复试畅通工程输入优化，并查集【1】

最新推荐文章于 2023-10-22 22:47:09 发布

qiang_____0712

最新推荐文章于 2023-10-22 22:47:09 发布

阅读量124

点赞数 3

CC 4.0 BY-SA版权

分类专栏： ACM 文章标签：输入优化并查集

本文链接：https://blog.youkuaiyun.com/qiang_____0712/article/details/87855596

ACM 专栏收录该内容

35 篇文章

订阅专栏

本文详细解析了如何使用最小生成树算法解决乡村公路建设问题，通过具体实例展示了算法的实现过程，包括输入输出样例、关键数据结构定义及算法流程说明。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://vjudge.net/problem/HDU-1233

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

Sample Output

3 5

Hint

Hint Huge input, scanf is recommended.

Source

浙大计算机研究生复试上机考试-2006年

#include<vector>
#include<algorithm>
#include<ctime>
#include<iostream>
#include<math.h>
using namespace std;
#define N 102
#define M N*(N-1)/2
#define INF 10000
typedef struct {
    int u,v,w;
} node;
node E[M];
int parent[N];
int iRank[N];

int cnt=0; //当变成n-1了才联通
bool cmp(node n1,node n2) {
    return n1.w<n2.w;
}
int root(int p);
int uni(int p, int q);
void InitUnion(int n) ;
int main() {
    int n,m;
    int kcase=0;
    while(scanf("%d",&kcase)&&kcase)
    {
        n=kcase;
        m=(n-1)*n/2;
        int uu,vv,ww;
        node T;

        for(int i=0; i<m; i++) {
            scanf ("%d%d%d",&E[i].u,&E[i].v,&E[i].w);
        }
        sort(E,E+m,cmp);
        int res=0;
        InitUnion(n);
        for(int i=0; i<m; i++)
        {

            T=E[i];
            uu=T.u;
            vv=T.v;
            if(uni(uu,vv)) {
                cnt++;
                res+=T.w;
                if(cnt==n-1) {
                    //res=min(res,E[j].w-E[i].w);
                    break;//一旦联通，j不再向后取了
                }
            }
        }
        cout<<res<<endl;
    }
    return 0;
}
void InitUnion(int n) {

    for(int i=0; i<n+1; i++) //初始化并查集
    {
        parent[i]=i;
        iRank[i]=2;
    }
    cnt=0;
}
int root(int p)
{
    if(p != parent[p])
    {
        parent[p] = root(parent[p]);
    }
    return parent[p];
}
int uni(int p, int q)
{
    int proot = root(p);
    int qroot = root(q);

    if(proot == qroot) return 0;

    if(iRank[proot] > iRank[qroot])
    {
        parent[qroot] = proot;
    }
    else
    {
        if(iRank[proot] = iRank[qroot])
        {
            iRank[qroot]++;
        }
        parent[proot] = qroot;
    }
    return 1;
}