Prim
城市公交网建设问题
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 5521 通过数: 2011
【题目描述】
有一张城市地图,图中的顶点为城市,无向边代表两个城市间的连通关系,边上的权为在这两个城市之间修建高速公路的造价,研究后发现,这个地图有一个特点,即任一对城市都是连通的。现在的问题是,要修建若干高速公路把所有城市联系起来,问如何设计可使得工程的总造价最少?
【输入】
n(城市数,1<≤n≤100)
e(边数)
以下e行,每行3个数i,j,wij,表示在城市i,j之间修建高速公路的造价。
【输出】
n-1行,每行为两个城市的序号,表明这两个城市间建一条高速公路。
【输入样例】
5 8
1 2 2
2 5 9
5 4 7
4 1 10
1 3 12
4 3 6
5 3 3
2 3 8
【输出样例】
1 2
2 3
3 4
3 5
开始UV【0~n】全为0表示顶点全未访问;
然后根据prim算法思想跟新UV[][]表;
U | v1,v2,v3,v4,v5,v6,v7 |
---|---|
V | v1,v2,v3,v4,v5,v6,v7 |
#include<iostream>
using namespace std;
#include <algorithm>
const int maxn = 101;
int G[maxn][maxn];
struct edge{//具有最小权值的边
int u, v;
int w;//权值
};
int UV[maxn][maxn];
void print(edge min) {
/*cout << "(u:" << min.u << "," << "v:" << min.v << "w:" << min.w << endl;*/
cout <<min.u << " " << min.v << endl;
}
int main() {
int n,e;//城市数,边数
cin >> n>>e;
for (int i = 0; i < e; i++) {
int u, v, w;
cin >> u >> v >> w;
G[u][v] = w;
G[v][u] = w;
}
UV[1][0] = 1;
UV[1][1] = 1;
int sum = 1;
for (int start = 1; start <= n; start++) {//n轮回
//cout << start << endl << endl;
edge min;
min.w = M; min.u = 0; min.v = 0;
for (int i = 1; i <= n; i++) {
if (UV[i][0] == 1) {
for (int j = 1; j <= n; j++) {
if (j == i)continue;
//cout << "UV[i][j]" << UV[i][j] << "G[i][j]" << G[i][j] << endl;
if (UV[j][1] == 0&&G[i][j]!=0) {
edge t;
t.u = i; t.v = j; t.w = G[i][j];
if (min.w > t.w)min = t;
//print(t);
}
}
}
}
print(min);
UV[min.v][0] = 1;
UV[min.v][1] = 1;
sum++;
if (sum == n)break;
}
return 0;
}