poj 2485 Highways

今天在做图论的题目。感觉图论的题目要不就不会做，要会做比较容易1a，因为坑人的地方比较少。

/*
POJ: 2485 Highways
*/
#include <iostream>
#include <cstdio>
#include <cstring>

#define MaxV 505
#define MaxL (1 << 20)

using namespace std;

long length[MaxV][MaxV];
int n;
int dis[MaxV];
bool used[MaxV];

long Prim();

int main()
{
int t;
//freopen("data.in", "rb", stdin);
scanf("%d", &t);
while(t--) {
scanf("%d", &n);
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
scanf("%d", &length[i][j]);
}
}

printf("%ld\n", Prim());
}
return 0;
}

long Prim()
{
long max_length = -1;

for(int i = 1; i < n; i++) {
used[i] = false;
}
used[0] = true;

dis[0] = 0;
for(int i = 1; i < n; i++) {
dis[i] = length[0][i];
}

for(int i = 0; i < n - 1; i++) {
long tmp = MaxL;
int tmp_index = 0;
for(int j = 0; j < n; j++) {
if(!used[j] && dis[j] < tmp) {
tmp = dis[j];
tmp_index = j;
}
}
used[tmp_index] = true;
max_length = max_length > tmp ? max_length : tmp;
for(int k = 0; k < n; k++) {
if(!used[k] && dis[k] > length[tmp_index][k]) {
dis[k] = length[tmp_index][k];
}
}
}

return max_length;
}