浙江财经大学
信工学院ACM集训队

HDU 1233 还是畅通工程(Kruskal)

本文由 Ocrosoft 于 2016-07-30 20:52:40 发表

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 39852    Accepted Submission(s): 18099

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

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

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

Sample Input
	
3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
 

Sample Output
	
3 5
Hint
Hint
Huge input, scanf is recommended.
 

Solution

Prime传送门:http://www.ocrosoft.com/?p=646

#include <set>
#include <map>
#include <list>
#include <cmath>
#include <stack>
#include <queue>
#include <ctime>
#include <string>
#include <cstdio>
#include <vector>
#include <cctype>
#include <climits>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define ms(a) memset(a,0,sizeof(a))
#define INF INT_MAX
#define MAXN 10000+10
using namespace std;
struct edge
{
    int u,v,w;
    bool f;
} edges[MAXN];
int n,m,cnt,sum,flag;
int pre[MAXN],son[MAXN];
void init()//初始化
{
    for(int i=1; i<=n; i++)pre[i]=-1;
}
bool cmp(edge a,edge b)
{
    return a.w<b.w;
}
int find(int x)
{
    if(pre[x]!=x)pre[x]=find(pre[x]);//路径压缩优化
    return pre[x];
}
bool Union(int x,int y)
{
    int fx=find(x),fy=find(y);
    pre[fx]=fy;
    return true;
    /*
    if(fx==fy)return false;//环
    else if(son[fx]>=son[fy])//加权优化
    {
        pre[fy]=fx;
        son[fx]+=son[fy];
    }
    else
    {
        pre[fx]=fy;
        son[fy]+=son[fx];
    }
    return true;
    */
}
int main()
{
    while(cin>>n&&n)
    {
        cnt=0,sum=0,flag=0;
        m=n*(n-1)/2;
        for(int i=1; i<=n; i++)
            pre[i]=i,son[i]=1;
        for(int i=1; i<=m; i++)
            scanf("%d%d%d",&edges[i].u,&edges[i].v,&edges[i].w);
        sort(edges+1,edges+m+1,cmp);
        for(int i=1;i<=m;i++)
        {
            int fx=find(edges[i].u),fy=find(edges[i].v);
            if(fx!=fy)pre[fx]=fy,sum+=edges[i].w;
            /*
            if(Union(edges[i].u,edges[i].v))
            {
                cnt++;
                sum+=edges[i].w;
                //printf("%d->%d\n",edges[i].u,edges[i].v);
            }
            if(cnt==n-1)//无解判断
            {
                flag=1;
                break;
            }
            */
        }
        printf("%d\n",sum);
    }
    return 0;
}

 

欢迎转载,请保留出处与链接。Ocrosoft » HDU 1233 还是畅通工程(Kruskal)

点赞 (0)or拍砖 (0)

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址