最小生成树问题,,做这道题做的好艰辛啊,,
第一阶段:没看题,看数据是标准的最短路径floyd算法的应用。。结果写出代码样例都过不了
第二阶段:看懂题,写kruskal算法,样例只能过一个,两个同样的输入输出的结果不一样。。。
第三阶段:解决了同输入不同输出的问题,【问题再次出现在kruskal算法的p数组的初始化上。节点的标号应该是1....n。】,但交上去wa!!!
第四阶段:功德圆满,解决了wa问题,终于修成正果,a掉了题目。【擦,问题又出现在了sort上,关于return a-b 和return a < b的问题,此处是间接排序,对整形数组排序,但真正参与比较的是double的数据】
代码如下:
#include <cstdio> #include <cmath> #include <cstring> #include <algorithm> using namespace std; #define N 755 #define M 300000 bool vis[N][N]; int n, m, t, dot[N][2]; int v[M], u[M], r[M], p[M]; double w[M]; void init() { m = 0; int a, b; double dx, dy; for(int i = 1; i <= n; i++) scanf("%d %d",&dot[i][0],&dot[i][1]); memset(vis,0,sizeof(vis)); scanf("%d",&t); for(int i = 0; i < t; i++) { scanf("%d%d",&a, &b); vis[a][b] = vis[b][a] = 1; } for(int i = 1; i <= n; i++) for(int j = i+1; j <= n; j++) { if(vis[i][j]) w[m] = 0.0; else { dx = dot[i][0]-dot[j][0]; dy = dot[i][1]-dot[j][1]; w[m] = sqrt(dx*dx+dy*dy); } u[m] = i; v[m] = j; m++; } } int cmp(const int i, const int j) { return w[i]<w[j]; } int find(int x) { return p[x]==x?x:p[x] = find(p[x]); } double kruskal() { double ans = 0; for(int i = 1; i <= n; i++) p[i] = i; for(int i = 0; i < m; i++) r[i] = i; sort(r,r+m,cmp); for(int i = 0; i < m; i++) { int e = r[i]; int x = find(u[e]); int y = find(v[e]); if(x!=y) { ans += w[e]; p[x] = y; } } return ans; } int main () { while(~scanf("%d",&n)) { init(); printf("%.2lf\n",kruskal()); } return 0; }