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

HDU 1879 继续畅通工程(Kruskal)

本文由 Ocrosoft 于 2016-07-30 17:11:23 发表

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 21755    Accepted Submission(s): 9304

Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
 

Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
 

Output
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
 

Sample Input

	
3 1 2 1 0 1 3 2 0 2 3 4 0 3 1 2 1 0 1 3 2 0 2 3 4 1 3 1 2 1 0 1 3 2 1 2 3 4 1 0
 

Sample Output

	
3 1 0
 

Author
ZJU
 

Solution

Prime算法传送门:http://www.ocrosoft.com/?p=671

#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%d",&edges[i].u,&edges[i].v,&edges[i].w,&edges[i].f);
        sort(edges+1,edges+m+1,cmp);
        for(int i=1;i<=m;i++)if(edges[i].f)Union(edges[i].u,edges[i].v);
        for(int i=1;i<=m;i++)
        {
            if(edges[i].f)continue;
            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 1879 继续畅通工程(Kruskal)

点赞 (0)or拍砖 (0)

评论 抢沙发

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