算法--最短关联连线

求连通所有水井的最短水管长度

输入含义：各个水井间距离如下所示

水井	A	B	C
A	0	1	4
B	1	0	2
C	4	2	0

input

2
3
0 1 4
1 0 2
4 2 0
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

output

#1 3
#2 28

code

#include <stdio.h>
int table[20][20];
int visit[20];
#define BIG_NUM 999999999
int main() {
  int i, j, t, T, N, result, min, r, step;
  scanf("%d", &T);
  for (t = 0; t < T; t++) {
    scanf("%d", &N);
    for (i = 0; i < N; i++) {
      for (j = 0; j < N; j++) {
        scanf("%d", &table[i][j]);
      }
      visit[i] = 0;
    }
    visit[0] = 1;  //设置第一个水井为入口
    result = 0;
    step = 1;
    while (step < N) {
      min = BIG_NUM;
      for (i = 0; i < N; i++) {
        //以标记过的水井为中心向周边寻找最短距离的水井
        if (visit[i] == 1) {
          for (j = 0; j < N; j++) {
            if (visit[j] == 0) {
              if (min > table[i][j]) {
                min = table[i][j];
                r = j;
              }
            }
          }
        }
      }
      if (min != BIG_NUM) {
        //将找到的最近的水井进行标记，并累加长度
        visit[r] = 1;
        result += min;
        step++;
      }
    }
    printf("#%d %d\r\n", t, result);
  }

  return 0;
}