首页 > 代码库 > POJ 1751 Highways

POJ 1751 Highways

也是最小生成树问题。输出剩余组成生成树的边。

Special Judge 答案不唯一。


不过很奇怪的是Kruskal 写的话。

C++提交要么刚好1000ms 要么就是TLE。

然后G++提交就是594ms。顺便求路过大神指点。


已经有很多的生成边了,用prim可能用时会少一点。


#include<cstdio>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<iostream>
#include<list>
#include<set>
#include<cmath>
#define INF 0x7fffffff
#define eps 1e-6
using namespace std;
int n,m;
int fa[751];
bool vis[751];
struct lx
{
    int u,v;
    double len;
} l[751*400];
struct node
{
    double x,y;
} point[751];
int father(int x)
{
    if(x!=fa[x])
        fa[x]=father(fa[x]);
    return fa[x];
}
bool cmpa(lx a,lx b)
{
    return a.len<b.len;
}
int main()
{
    scanf("%d",&n);
    for(int i=0; i<=n; i++)
        fa[i]=i;
    for(int i=1; i<=n; i++)
        scanf("%lf%lf",&point[i].x,&point[i].y);
    int cot=0;
    for(int i=1; i<=n; i++)
        for(int j=i+1; j<=n; j++)
        {
            l[cot].u=i;
            l[cot].v=j;
            l[cot++].len=sqrt(pow(point[i].x-point[j].x,2)+pow(point[i].y-point[j].y,2));

        }
    sort(l,l+cot,cmpa);
    scanf("%d",&m);
    while(m--)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        int fu=father(u);
        int fv=father(v);
        fa[fv]=fu;
    }
    int num=0;
    for(int i=0; i<cot; i++)
    {
        int fu=father(l[i].u);
        int fv=father(l[i].v);
        if(fu==fv)continue;
        fa[fv]=fu;
        num++;
        printf("%d %d\n",l[i].u,l[i].v);
        if(num>n-1)break;
    }
}