zoj 1586 QS Network

最新推荐文章于 2017-03-25 01:06:00 发布

原创最新推荐文章于 2017-03-25 01:06:00 发布 · 343 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#zoj #prim

本文介绍了一种使用Prim算法来解决QS网络最小费用构建问题的方法。通过初始化网络节点及边的权重，并考虑适配器成本，实现了寻找连接所有节点的最小生成树。代码示例详细展示了Prim算法的实现过程。

题意：求建立QS网络的最小费用。

思路：prim。

代码：

#include <iostream>
#include <climits>

using namespace std;

const int maxn = 1005;

int n;
int edge[maxn][maxn];
int adapter[maxn];
int lowcost[maxn];

void Init(){
	cin>>n;
	for(int i = 0; i < n; i++)
		cin>>adapter[i];
	for(int i = 0; i < n; i++){
		for(int j = 0; j < n; j++){
			cin>>edge[i][j];
			if(i == j) edge[i][j] = INT_MAX;
			else edge[i][j] += adapter[i] + adapter[j];
		}
	}
	for(int i = 0; i < n; i++)
		lowcost[i] = INT_MAX;
}

int Prim(int u){
	int sum = 0;
	for(int i = 0; i < n; i++)
		lowcost[i] = edge[u][i];
	lowcost[u] = -1;
	
	for(int i = 0; i < n-1; i++){
		int min = INT_MAX;
		int v;
		
		for(int j = 0; j < n; j++){
			if(lowcost[j] != -1 && lowcost[j] < min){
				v = j; min = lowcost[j];
			}
		}
		
		sum += min;
		lowcost[v] = -1;
		
		for(int j = 0; j < n; j++){
			if(edge[v][j] < lowcost[j])
				lowcost[j] = edge[v][j];
		}
		
	}//end of for
	return sum;
}

int main(){
	int t;
	cin>>t;
	while(t--){
		Init();
		cout<<Prim(0)<<"\n";
	}
	return 0;
}