继续畅通工程

题目描述

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

输入描述:

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

    当N为0时输入结束。

输出描述:

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

示例1

输入

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

//在这道题目当中已经给出了修建好的边，本质仍然是求最小生成树权值
//问题，但是已经修建好的路的优先级应该高于未修建好的（在边进行排序的时候）

//继续畅通工程给出的道路有的是已经修建好的，如果优先使用已经修建好的那么最终会
//花费的少一点，即在优先使用修建好道路的基础上求最小生成树
#include<iostream>
#include<algorithm>
using namespace std;
struct E
{
    int start,end;
    int cost;
    int finish;
}edges[1000];
int pre[1000];
int find(int x)
{
    if(pre[x]==-1)return x;
    else return find(pre[x]);
}
bool cmp(E a,E b)
{
    if(a.finish!=b.finish)return a.finish>b.finish;
    else if(a.finish==b.finish)return a.cost<b.cost;
}
int main()
{
    int n;
    while(cin>>n&&n!=0)
    {
        for(int i=1;i<=n;i++)
            pre[i]=-1;
        int m=n*(n-1)/2;
        for(int i=0;i<m;i++)
        {
            cin>>edges[i].start>>edges[i].end>>edges[i].cost>>edges[i].finish;
        }
        sort(edges,edges+m,cmp);
        int ans=0;
        for(int i=0;i<m;i++)
        {
            int t1=find(edges[i].start);
            int t2=find(edges[i].end);
            if(t1!=t2)
            {
                pre[t1]=t2;
                if(edges[i].finish==0)
                      ans+=edges[i].cost;
            }
        }
        cout<<ans<<endl;
    }
}