HDU 1879 继续畅通工程

继续畅通工程

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

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

 

Source

浙大计算机研究生复试上机考试-2008年

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;

const int MAXN = 105;
const int INF = 1e9;

vector<pair<int,int> > pMap[MAXN];

int prim(int N);

int main(){
    int N;

    while(scanf("%d",&N) && N){

        for(int i = 0; i < MAXN; i ++)
            pMap[i].clear();

        int M = N * (N - 1) /2 ;
        int u,v,w,tag;
        for(int i = 0; i < M; i ++){
            scanf("%d%d%d%d",&u,&v,&w,&tag);
            if(tag){
                pMap[u].push_back(make_pair(v,0));
                pMap[v].push_back(make_pair(u,0));
            }else{
                pMap[u].push_back(make_pair(v,w));
                pMap[v].push_back(make_pair(u,w));
            }
        }
        cout << prim(N)<<endl;
    }
    return 0;
}
int prim(int N){

	int nCost = 0;
//	vector<int> pMST;
	int pCost[MAXN];
//	pMST.push_back(1);
	pCost[1] = -1;
	for(int i = 2; i <= N; i++){
        pCost[i] = INF;
    }
	for(int i = 0; i < pMap[1].size(); i++){
        pCost[pMap[1][i].first] = pMap[1][i].second;
    }
	for(int i = 1; i <= N - 1; i++){
		int nVertex = 0, nWeight = INF;
		for(int j = 1; j <= N; j++){
			if(nWeight > pCost[j] && pCost[j] != -1){
				nVertex = j;
				nWeight = pCost[j];
			}
		}

		pCost[nVertex] = -1;
//		pMST.push_back(nVertex);

		nCost += nWeight;

		for(int j = 0; j < pMap[nVertex].size(); j++){
			if(pCost[pMap[nVertex][j].first] != 0 &&
				pCost[pMap[nVertex][j].first] > pMap[nVertex][j].second){
				pCost[pMap[nVertex][j].first] = pMap[nVertex][j].second;
			}
		}
	}
//	cout << "MST Cost is " << nCost << endl;
//	cout << "The vertexs in MST are ";
//	for(int i = 0; i < pMST.size(); i++){
//        cout << pMST[i] << " ";
//    }
//	cout << endl;

	return nCost;
}