题目1028:继续畅通工程

来源:互联网 发布:西安软件开发工资待遇 编辑:程序博客网 时间:2024/06/08 03:29
题目描述:
    省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
输入:
    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。

    当N为0时输入结束。
输出:
    每个测试用例的输出占一行,输出全省畅通需要的最低成本。
样例输入:
31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10
样例输出:
31

0

C++代码:

#include<iostream>#include<vector>#include<algorithm>using namespace std;int n;int v[100];struct node{    int a;    int b;    int val;};vector<node> ns;bool cmp(node a,node b){    if(a.val<b.val)        return true;    return false;}int main(){    while(cin>>n&&n!=0){        int sum=0;        for(int i=0;i<n*(n-1)/2;i++){            int a,b,c,d;            cin>>a>>b>>c>>d;            if(d==0){                node nd;                nd.a=a;                nd.b=b;                nd.val=c;                ns.push_back(nd);            }else{                v[a]=1;                v[b]=1;            }        }        sort(ns.begin(),ns.end(),cmp);        int t=0;        for(int i=1;i<=n;i++){            if(v[i]==1)                t++;        }        while(t<n){            for(int i=0;i<ns.size();i++){                if(!(v[ns[i].a]==1&&v[ns[i].b]==1)){                    sum+=ns[i].val;                    if(v[ns[i].a]==0&&v[ns[i].b]==0){                        t+=2;                    }                    if(v[ns[i].a]==0&&v[ns[i].b]!=0||v[ns[i].a]!=0&&v[ns[i].b]==0){                        t=t+1;                    }                    v[ns[i].a]=1;                    v[ns[i].b]=1;                    break;                }            }        }        cout<<sum<<endl;        ns.clear();        for(int i=1;i<=n;i++)            v[i]=0;    }    return 0;}