[kuangbin带你飞]专题六 最小生成树--poj--1751(kruskal)

H - Highways POJ - 1751

思路：

本题也有一些已经建好的边。

1. 因为点数v有750个，换算一下边数就有750*750个了，kruskal可能会超时。（sort的时间和边数有关）；
2. 读入时，把已经建好的边，连起来，
3. 对于一些已经建好的边，就不要再建边了（并查集基本操作：每建一条边时，判断这两个点，是否在一个连通块里，在，就不连了，）

反思

1. 由于这道题kruskal，可能会被卡，所以我在找其他xaijib优化。
2. 最发现写bool operator比cmp快许多，以后就bool operator了（香）

struct EDGE
{
    int from, to, val, next;
    bool operator <(const EDGE&a)const//this*******
    {return val<a.val;}//this*********
    EDGE(){}
    fzhead:fzbody{}
}e[maxm];

AC

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define For(i,x,y) for(register int i=(x); i<=(y); i++)
#define mst(x,a) memset(x,a,sizeof(x))
#define fzhead EDGE(int _from, int _to, int _val, int _next)
#define fzbody from(_from),to(_to), val(_val), next(_next)
using namespace std;
typedef long long ll;
const int maxn=750+10;
const int maxm=1e6+10;
struct EDGE
{
    int from, to, val, next;
    bool operator <(const EDGE&a)const
    {return val<a.val;}
    EDGE(){}
    fzhead:fzbody{}
}e[maxm];
struct point
{int x,y;}p[maxn];
int cnt,vis[maxn],head[maxn],f[maxn];
void add(int bg, int to, int val)
{
    e[++cnt]=EDGE(bg,to,val,head[bg]);
    head[bg]=cnt;
}
int find(int x)
{
    if(x==f[x])return f[x];
    return f[x]=find(f[x]);
}
//bool cmp(EDGE a,EDGE b)
//{return a.val<b.val;}
int get_dis(point a, point b)
{return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);}
int main()
{
    int n,m,tot=1,a,b,q;
    scanf("%d", &n);
    mst(head,-1);
    For(i,1,n)f[i]=i;
    For(i,1,n)scanf("%d%d",&p[i].x,&p[i].y);
    scanf("%d", &q);
    For(i,1,q)
    {
        scanf("%d%d",&a,&b);
        a=find(a);b=find(b);
        if(a!=b){f[a]=b;tot++;}
    }
    For(i,1,n)
    {
        For(j,i+1,n)
        {
            a=find(i);b=find(j);
            if(a==b)continue;
            add(i,j,get_dis(p[i],p[j]));
        }
    }
    sort(e+1,e+1+cnt);
   // ll ans=0;
    For(i,1,cnt)
    {
        a=find(e[i].from);b=find(e[i].to);
        if(a==b)continue;
        f[a]=b;
        tot++;
    //    ans+=e[i].val;
        printf("%d %d\n",e[i].from,e[i].to);
        if(tot==n)break;
    }
    //printf("%lld\n",ans);
    return 0;
}