uva10397 - Connect the Campus(连接校园)

最小生成树问题,,做这道题做的好艰辛啊,,

第一阶段:没看题,看数据是标准的最短路径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;
}


你可能感兴趣的:(uva10397 - Connect the Campus(连接校园))