1118 继续畅通工程

1118: 继续畅通工程

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

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

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

• 输入示例

  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
  

• 输出示例

  3
  1
  0
  

• 代码块

  #include<bits/stdc++.h>
  using namespace std;
  struct edge{//边结点
      int x, y;//两邻接点
      int w;//权值
  };
  
  const int N = 101;
  int fa[N];//查并集，用于判断点是否在同一联通分量里面
  
  int get(int x){
      return x==fa[x]?x:fa[x]=get(fa[x]);
  }
  
  void Kruskal(vector<edge> e, int n){//假设已经排好序, n表示点的数目
      for(int i=1; i<=n; i++) fa[i]=i;//查并集初始化
      int ans = 0;
      for(int i=0; i<e.size(); i++){//遍历所有的边，假设边的权值已经排好序
          int fx = get(e[i].x);
          int fy = get(e[i].y);//分别得到正在遍历的边的两个端点
          if(fx != fy){//这两个端点没有连通,需要合并
              ans = ans + e[i].w;//加上这条边的权值（说明当前边已经选中）
              fa[fx] = fy;//合并（这条边的两个端点加入到同一个连通分量）
          }
      }
      printf("%d\n", ans);
  }
  
  int main(){
      int i, n;
      while(~scanf("%d", &n)){
          if(n == 0) break;
          int m = n*(n-1)/2;//边的数量
          vector<edge> e(m);
          for(i=0; i<m; i++){
              scanf("%d%d%d", &e[i].x, &e[i].y, &e[i].w);
              int p;//表示修建的状态
              scanf("%d", &p);
              if(p==1) e[i].w=0;//这条边已经修过了，代价为0
          }
          sort(e.begin(), e.end(), [](edge a, edge b)->bool{return a.w<b.w;});//排序，目前不需要掌握
          Kruskal(e, n);
      }
      system("pause");
      return 0;
  }