hdu1875 畅通工程再续--prim

来源：互联网发布：红蜘蛛软件路由器编辑：程序博客网时间：2024/06/04 19:23

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

其实这题很简单，但是当时脑子不知道怎么了，两点不连通我把它表示成-1标记，而不是直接赋值为最大值，导致接下来的操作有点混乱。

代码：

#define _CRT_SECURE_NO_DEPRECATE #include<iostream>#include<cmath>#include<algorithm>using namespace std;double x[105], y[105];double a[105][105];double lowCost[105];int T;int C;int num;//标记是否是所有点都加入到生成树中double getDis(int i, int j){return sqrt((x[i] - x[j])*(x[i] - x[j]) + (y[i] - y[j])*(y[i] - y[j]));}double prim(){double ans = 0;num = 1;//起点默认是加入的，所以是1for (int i = 1; i <= C; i++)lowCost[i] = a[1][i];//默认1为起点lowCost[1] = 0;//标记起点已被加入生成树for (int i = 1; i < C; i++){double min = 99999999;int minIndex;for (int i = 1; i <= C; i++){if (lowCost[i] != 0 && lowCost[i] != -1 && lowCost[i] < min){min = lowCost[i];minIndex = i;}}if (min == 99999999)//如果无法构成所有节点的生成树，就会提前结束return ans;num++;//又加入一个ans += min;lowCost[minIndex] = 0;for (int i = 1; i <= C; i++){if (lowCost[i] != 0 && a[minIndex][i] != -1 && a[minIndex][i] < lowCost[i] || lowCost[i] == -1)lowCost[i] = a[minIndex][i];}}return ans;}int main(){scanf("%d", &T);while (T--){scanf("%d", &C);for (int i = 1; i <= C; i++)scanf("%lf%lf", &x[i], &y[i]);if (C <= 1)printf("oh!\n");else{for (int i = 1; i <= C; i++){for (int j = 1; j <= C; j++){double dis = getDis(i, j);if (dis < 10 || dis>1000)a[i][j] = -1;//-1表示该点不通elsea[i][j] = dis;}}double ans = prim();if (num != C)printf("oh!\n");elseprintf("%.1lf\n", ans * 100);}}return 0;}

1 0