【HPU OJ 1409 】Watch Dog 【最小生成树】

题目描述
在2050年某城市，一个神秘组织通过一种名叫ctOS的控制系统掌控了全城所有人的联网设备（手机、电脑、监控、交通信号灯，甚至是银行警察局的安保系统），该系统通过很多遍布城市各地的基站来完成对电子设备的控制。

Aiden Pearce（艾登·皮尔斯）是一名超级黑客，因他叛出该组织而被组织报复，谋划了一场”车祸”害死了他的女儿。艾登为了给女儿报仇，决心铲除该组织。

首先他要先侵入stOS系统获取该组织的资料，然而这些资料被分散存到了所有的基站中，所以他要用自己改装的手机获取所有基站的访问权。他的手机电量有限，如果电量为0以下，则无法继续进行工作。幸运的是，艾登可以通过已经侵入的基站，将其作为傀儡机，继续入侵其它的基站。所消耗的电量为两个设备（手机~基站、基站~基站）距离的平方。

举个例子，艾登的初始位置为(0,0)，有两个基站的位置分别为(1,0)(0,2)，艾登可以先从他的位置入侵第一个基站，耗电为1；再通过第一个基站，入侵第二个基站，耗电为5。总耗电为6。

输入
有多组测试数据。

每组数据第一行输入两个整数n和k，n表示一共有n个基站，k表示现在手机的电量。

接下来一行输入两个整数p,q。表示艾登现在的位置。

接下来的n行，每行输入两个两个整数：Xi,Yi。表示第i个基站的位置。

1≤n≤5001≤n≤500
0≤k≤1090≤k≤109
−200≤x,y,Xi,Yi≤200−200≤x,y,Xi,Yi≤200
输出
对于每组数据，输出占一行。

如果艾登可以入侵所有的基站，输出：”Go ahead!”（不带引号）

如果艾登不能入侵所有的基站，输出：”Go a head!”（不带引号）

样例输入
2 6
0 0
1 0
0 2
3 5
0 0
1 1
1 -1
-1 1
样例输出
Go ahead!
Go a head!

哎，就是模板
代码

#include<cstring>
#include<algorithm>
#include<cstdio>
#define LL long long
using namespace std;

const int MAXN =500+10;
const int MAXM = 1e6+100;
int n,k;

struct  Node{
    int x,y;
}node[MAXN];
struct Edge {
    int from,to,val;
}edge[MAXM];

int getdis(Node a,Node b){
    return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);
}
int pre[MAXN];
void init(){
    for(int i=0;i<=n+10;i++)   pre[i]=i;
} 
int Find(int x){
    return x==pre[x]?x:pre[x]=Find(pre[x]);
}
void join(int a,int b){
    a=Find(a);b=Find(b);
    if(a!=b) pre[a]=b;
}
bool cmp(Edge a,Edge b){
    return a.val<b.val;
}

int main(){
    while(~scanf("%d%d",&n,&k)){
        int size=1;init();
         int x,y;scanf("%d%d",&x,&y);
         node[size++]={x,y};
         for(int i=1;i<=n;i++){
            scanf("%d%d",&x,&y);
            node[size++]={x,y};
         } 
         int sz=0;
         for(int i=1;i<size;i++){
            for(int j=i+1;j<size;j++){
                edge[sz++]={i,j,getdis(node[i],node[j])};
             }
         }

         sort(edge,edge+sz,cmp);
//       for(int i=0;i<sz;i++) {
//          printf("%d %d  %d\n",edge[i].from,edge[i].to,edge[i].val);
//       }
         int cnt=0;int i;
         for(i=0;i<sz;i++){
            Edge e=edge[i];
            if(Find(e.from)!=Find(e.to)){
                cnt++;
                k-=e.val;
                join(e.from,e.to);
              }
         }
         if(cnt==n&&k>=0) puts("Go ahead!");
         else puts("Go a head!");
     }
    return 0;
}