HDOJ 1879 继续畅通工程

 

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1879

 

源代码：

//HDOJ 1879 AC
#include <iostream>
#include <cmath>

using namespace std;
#define MAX 99999
#define LEN 101
int dist[LEN];
int map[LEN][LEN];
bool isvisited[LEN];

//初始化
void init(){
	int i,j;
	for(i=0;i<LEN;i++){
		for(j=0;j<LEN;j++){
			if(i==j) map[i][j]=0;   //对a[][]进行初始化，一般都要；
			map[i][j]=MAX;
		}
	}
}


//prim算法
int prim(int n){
	int i,j,min,pos;
	int sum=0;
	memset(isvisited,false,sizeof(isvisited));

	//初始化
	for(i=1;i<=n;i++){
		dist[i]=map[1][i];
	}

	//从1开始
	isvisited[1]=true;
	dist[1]=MAX;

	//找到权值最小点并记录下位置
	for(i=1;i<n;i++){
		min=MAX;
		//pos=-1;
		for(j=1;j<=n;j++){
			if(!isvisited[j] && dist[j]<min){
				min=dist[j];
				pos=j;
			}
		}	
		sum+=dist[pos];//加上权值
		isvisited[pos]=true;

		//更新权值
		for(j=1;j<=n;j++){
			if(!isvisited[j] && dist[j]>map[pos][j]){
				dist[j]=map[pos][j];
			}
		}
	}	
	return sum;
}

int main(){

	int n;

	while(cin>>n){
		init();
		if(n==0) break;

		int i,j,a,b,d,state;

		for(i=0;i<n*(n-1)/2;i++){

			cin>>a>>b>>d>>state;

			if(state==1){
				map[a][b]=map[b][a]=0;
			}
			else{
				map[a][b]=map[b][a]=d;
			}
		}
		cout<<prim(n)<<endl;

	}	
	 return 0;
}