HPUoj 1409: Watch Dog [最小生成树]

Watch Dog

Description

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

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

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

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

Input

有多组测试数据。

每组数据第一行输入两个整数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

Output

对于每组数据，输出占一行。

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

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

Sample Input

2 6
0 0
1 0
0 2
3 5
0 0
1 1
1 -1
-1 1

Sample Output

Go ahead!
Go a head!

题解：

最小生成树模板 脑抽了 比赛时候去跑了了个最短路 mdzz

AC代码

#include <cstdio>
#include <cmath>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

#define LL long long
#define CLR(a,b) memset(a,(b),sizeof(a))

const int INF = 0x3f3f3f3f;
const LL INFLL = 0x3f3f3f3f3f3f3f3f;
const int N = 5e6+10;
int n, zz;
struct node {
    int x, y;
    int val;
}p[N];
int xrr[N], yrr[N];
int par[N];
int distt(int x1, int x2, int y1, int y2)
{
    return (x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
}
bool cmp(node x, node y)
{
    return x.val < y.val;
}
void init()
{
    for(int i = 0; i <= n+1; i++) par[i] = i;
}
int find(int x)
{
    if(x == par[x]) return x;
    return par[x] = find(par[x]);
}
bool unite(int x,int y)
{
    x = find(x);
    y = find(y);
    if(x != y) {
        par[y] = x; return true;
    }
    return false;
}
int main()
{
    while(~scanf("%d%d",&n,&zz)) {
        init();
        for(int i = 1;i <= n+1; i++) {
            scanf("%d%d",&xrr[i],&yrr[i]);
        }
        int k = 0;
        for(int i = 1; i <= n+1; i++) {
            for(int j = i+1; j <= n+1; j++) {
                p[k].x = i; p[k].y = j;
                p[k++].val = distt(xrr[i],xrr[j],yrr[i],yrr[j]);
            }
        }
        int sum = 0, res = 0;
        sort(p, p+k, cmp);
        for(int i = 0; i < k; i++) {
            if(unite(p[i].x,p[i].y)) {
                sum += p[i].val;
                res++;
            }
            if(res == n) {
                 break;
            }
        }
        if(sum > zz) puts("Go a head!");
        else puts("Go ahead!");
    }
return 0;
}