4941-继续畅通工程(最小生成树 克鲁斯卡尔算法) ZCMU

Description

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

Input

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

 当N为0时输入结束。

Output

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

Sample Input

4

1 2 1 1

1 3 6 0

1 4 2 1

2 3 3 0

2 4 5 0

3 4 4 0

3 1 2 1

1 2 3 2

1 1 3 1

0 0

Sample Output

3

0

思路

题目中的给出每条路的修建状态这一条件我们可以进行转化，若一条路已修建，那么就相当于这条路的修建成本为0；

代码

#include<bits/stdc++.h>
#define MAX 10005
using namespace std;
struct edge{
  int u;
  int v;
  int w;
  int state;
}e[MAX];
bool cmp(edge x,edge y)
{
    return x.w<y.w;
}
int n,m,f[MAX],sum,c;
int getf(int v)
{
    if(f[v]==v)
        return v;
    else
    {
        f[v]=getf(f[v]);
        return f[v];
    }
}
int Merge(int v,int u)
{
    int t1=getf(v);
    int t2=getf(u);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}
void init()
{
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
    }
}
int main()
{
    int i;
    while(~scanf("%d",&n))
    {
      if(n==0) break;
      if(n==1)
      {
         printf("0\n");
         continue;
      }
      int m=n*(n-1)/2,x;
      for(i=1;i<=m;i++)
      {
        scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].w,&e[i].state);
        if(e[i].state) e[i].w=0;
      }
      sort(e+1,e+m+1,cmp);
      init();
      c=sum=0;
      for(i=1;i<=m;i++)
      {
        if(Merge(e[i].u,e[i].v))
         {
             c+=1;
             sum+=e[i].w;
         }
         if(c==n-1)
            break;
      }
      printf("%d\n",sum);
    }
    return 0;
}