Codeforces Problemset 198E(#125 div.1 E)

问题描述

　　在一个二维平面上，你现在的位置在（x，y）同时你手上有一块磁铁。
　　在这个平面上，还有N块散落的磁铁，每个磁铁都可以抽象成一个点，你的目标是吸引最多的散落的磁铁。
　　每一块磁铁都有五个属性，x，y，m，p，r，分别表示磁铁的横坐标，磁铁的纵坐标，磁铁的重量，磁铁的吸引力，磁铁的吸引半径。
　　一块磁铁想要把另一块磁铁吸过来的条件，有两条。
　　1.被吸引的磁铁和吸引的磁铁之间的距离小于等于吸引磁铁的吸引半径。这里距离计算的是欧几里得距离。
　　2.被吸引的磁铁的重量小于等于吸引磁铁的吸引力。
　　任何被你吸过来的磁铁都可以用来吸引新的磁铁。每块磁铁可以吸引无数多次，但是每次只能有一块磁铁在吸引，不能多块同时吸引。同时你的位置（x，y）也是不变的。
　　现在你想要知道，你最多可以吸引多少散落的磁铁。

输入格式

　　第一行有五个整数，x，y，p，r，N。其中x，y，p，r分别表示你的坐标和你一开始拥有的磁铁的吸引力，吸引半径。N表示散落的磁铁数目。
　　接下来N行每行5个整数，xi，yi，mi，pi，ri，依次描述第i块散落的磁铁的横坐标，纵坐标，重量，吸引力，吸引半径。

输出格式

　　输出一行，包含一个整数，你最多可以吸引的散落的磁铁数目。

样例输入

0 0 5 10 5
5 4 7 11 5
-7 1 4 7 8
0 2 13 5 6
2 -3 9 3 4
13 5 1 9 9

样例输出

3

数据规模和约定

　　N<=250000
　　-10 ^ 9 <= 坐标x,y <= 10 ^ 9
　　1 <= 所有的m,p,r <= 10 ^ 9
　　输入数据保证不含有任何两块磁铁在同一个位置。

显然，线段树套平衡树……

纯粹纪念一下set……线段树套平衡树才写了1.5k

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <set>
using namespace std;
int i,j,k,X,Y,all,tot,n,M[250011],P[250011],R[250011],Left[500011],Right[500011],PP[250011],dl[250011],sum[500011];
long long x,y,p,r,D[250011],RR[250011];
bool used[250011];
set<pair<int,int> > ent[500011];
bool cmp(int a,int b)
{
		 return D[a]<D[b];
}
void build(int s,int e,int now)
{
		 sum[now]=e-s+1;
		 for (int i=s;i<=e;i++)
		 		 ent[now].insert(pair<int,int>(M[dl[i]],dl[i]));
		 if (s==e) return ;
		 int mid=(s+e)/2;
		 Left[now]=++tot;
		 build(s,mid,tot);
		 Right[now]=++tot;
		 build(mid+1,e,tot);
}
void quick(long long A,int b,int l,int r,int now)
{
		 if (sum[now]==0 || D[dl[l]]>A) return ;
		 if (D[dl[r]]<=A)
		 {
		 		set<pair<int,int> >::iterator cp;
		 		while (sum[now] && ent[now].begin()->first<=b)
		 		{
							cp=ent[now].begin();
						 	int o=cp->second;
							if (!used[o])
							{
								 used[o]=true;
								 PP[++all]=P[o];
								 RR[all]=(long long)R[o]*R[o];
							}
							ent[now].erase(cp);
							sum[now]--;
		 		}
		 		return ;
		 }
		 int mid=(l+r)/2;
		 quick(A,b,l,mid,Left[now]);
		 if (D[dl[mid+1]]<=A) quick(A,b,mid+1,r,Right[now]);
}
int main()
{
		cin >> x >> y >> p >> r >> n;
		for (i=1;i<=n;i++) 
				scanf("%d%d%d%d%d",&X,&Y,&M[i],&P[i],&R[i]),D[i]=(X-x)*(X-x)+(Y-y)*(Y-y),dl[i]=i;
		sort(dl+1,dl+n+1,cmp);
		build(1,n,0);
		PP[0]=p;
		RR[0]=r*r;
		for (i=0;all<=n && i<=all;i++)
				quick(RR[i],PP[i],1,n,0);
		cout << all;
		return 0;
}