vijos P1810导弹拦截 （排序+树状数组）

P1810导弹拦截

Accepted

标签： NOIP普及组2010 [显示标签]

描述

经过11 年的韬光养晦，某国研发出了一种新的导弹拦截系统，凡是与它的距离不超过其工作半径的导弹都能够被它成功拦截。当工作半径为0 时，则能够拦截与它位置恰好相同的导弹。但该导弹拦截系统也存在这样的缺陷：每套系统每天只能设定一次工作半径。而当天的使用代价，就是所有系统工作半径的平方和。 某天，雷达捕捉到敌国的导弹来袭。由于该系统尚处于试验阶段，所以只有两套系统投入工作。如果现在的要求是拦截所有的导弹，请计算这一天的最小使用代价。

格式

输入格式

第一行包含4 个整数x1、y1、x2、y2，每两个整数之间用一个空格隔开，表示这两套导弹拦截系统的坐标分别为(x1, y1)、(x2, y2)。

第二行包含1 个整数N(1 ≤ N ≤ 100000)。表示有N 颗导弹。
接下来N 行，每行两个整数x、y，中间用一个空格隔开，表示一颗导弹的坐标(x, y)。不同导弹的坐标可能相同。
所有坐标分量的绝对值都不超过1000。

输出格式

只有一行，包含一个整数，即当天的最小使用代价。

样例1

样例输入1[复制]

0 0 10 0 
2 
-3 3  
10 0 

样例输出1[复制]

18

样例2

样例输入2[复制]

0 0 6 0 
5  
-4 -2 
-2 3    
4 0 
6 -2 
9 1

样例输出2[复制]

30

限制

每个测试点1s。

提示

两个点(x1, y1)、(x2, y2)之间距离的平方是(x1− x2)2+(y1−y2)2。

两套系统工作半径r1、r2 的平方和，是指r1、r2 分别取平方后再求和，即r12 +r22 。
样例1说明：
样例1 中要拦截所有导弹，在满足最小使用代价的前提下，两套系统工作半径的平方分 别为18 和0。
样例2说明：
样例中的导弹拦截系统和导弹所在的位置如下图所示。要拦截所有导弹，在满足最小使用代价的前提下，两套系统工作半径的平方分别为20 和10。

来源

NOIP2010普及组

题解：排序+树状数组

计算所有点到第一个拦截系统的距离，按照从小到大排序，然后枚举距离第一个拦截系统最远的点，那么剩下的点都归第二个拦截系统拦截，统计剩下点到第二个拦截系统的最大值，然后相加更新答案。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define N 100003
using namespace std;
int n,m;
struct data {
	int dis,dis1,x,y;
}a[N];
int x1,x2,y11,y2,tr[N];
int pow1(int x)
{
	return x*x;
}
int cmp(data a,data b)
{
	return a.dis<b.dis;
}
int lowbit(int x)
{
	return  x&(-x);
}
void change(int v,int x)
{
	for (int i=x;i<=n;i+=lowbit(i)) tr[i]=max(tr[i],v);
}
int find(int x)
{
	int ans=0; if (x==0) return 0;
 	for (int i=x;i>=1;i-=lowbit(i))
	  ans=max(ans,tr[i]);
	return ans; 
}
int main()
{
	scanf("%d%d%d%d",&x1,&y11,&x2,&y2);
	scanf("%d",&n);
	for (int i=1;i<=n;i++){
		scanf("%d%d",&a[i].x,&a[i].y);
		a[i].dis=pow1(a[i].x-x1)+pow1(a[i].y-y11);
		a[i].dis1=pow1(a[i].x-x2)+pow1(a[i].y-y2);
	}
	sort(a+1,a+n+1,cmp);
	for (int i=n;i>=1;i--) change(a[i].dis1,n-i+1);
	int ans=1000000000;
	for (int i=1;i<=n;i++) {
		int t=a[i].dis+find(n-i);
		ans=min(ans,t);
	}
	printf("%d\n",ans);
 }