#广搜，分块#jzoj 3974 CH #46A 磁力块

题目

主人公一开始在一个位置用磁铁吸引其它磁铁，当磁铁的质量$\le$磁力就会吸到主人公手里，不在主人公手里的磁铁不会互相吸引，主人公可以更换磁铁，但是只能原地不动，问主人公最多能吸到多少块磁铁

---

分析

这道题可以用广搜做，必然的暴力，但是问题是直接暴力会超时，怎么办？分块优化，用分块得到一个个区间，当然区间之间的排序用距离排序，区间内的排序用重量排序，然后，就没有然后了

---

代码

#include <cstdio>
#include <cmath>
#include <algorithm>
#include <queue>
typedef long long ll; std::queue<int>q; bool v[250001];
struct rec{ll dis,rad; int m,p;}a[250001];
int X0,Y0,n,x,y,tot,l[511],r[511],ans; ll d[511];
ll in(){
	ll ans=0; int f=1; char c=getchar();
	while ((c<48||c>57)&&c!='-') c=getchar();
	if (c=='-') c=getchar(),f=-f;
	while (c>47&&c<58) ans=ans*10+c-48,c=getchar();
	return ans*f;
}
int min(int a,int b){return (a<b)?a:b;}
bool cmp1(const rec &a,const rec &b){return a.dis<b.dis;}
bool cmp2(const rec &a,const rec &b){return a.m<b.m;}
int main(){
	X0=in(); Y0=in(); a[0].p=in(); a[0].rad=in(); n=in(); a[0].rad*=a[0].rad;//以便比较
	for (register int i=1;i<=n;i++){
		x=in(); y=in(); a[i].m=in(); a[i].p=in(); a[i].rad=in();
		a[i].rad*=a[i].rad; a[i].dis=((ll)X0-x)*((ll)X0-x)+((ll)Y0-y)*((ll)Y0-y);//直线距离，为了精度，没开方
	}
	std::stable_sort(a+1,a+1+n,cmp1);
	for (register int i=1;i<=n;i+=500){
		l[++tot]=i; r[tot]=min(n,i+499);//区间左端点，区间右端点
		d[tot]=a[r[tot]].dis; std::stable_sort(a+l[tot],a+r[tot]+1,cmp2);//稍微排个序
	}
	q.push(0);
	while (q.size()){
		ll rr=a[q.front()].rad; int pp=a[q.front()].p; q.pop();
		for (register int i=1;i<=tot;i++){
			if (d[i]>rr){//距离不够（可能在里面）
				for (register int j=l[i];j<=r[i];j++)
				    if (!v[j]&&a[j].dis<=rr&&a[j].m<=pp) v[j]=1,ans++,q.push(j);//每吸到一块磁铁增加1
				break;
			}
			while (l[i]<=r[i]&&a[l[i]].m<=pp){//吸吸吸吸
				if (!v[l[i]]) ans++,q.push(l[i]);
				l[i]++;
			}
		}
	}
	printf("%d",ans);
    return 0;
}