抓企鹅

题目描述

Xyz带着他的教徒们乘着科考船一路破冰来到了南极大陆，发现这里有许许多多的企鹅。邪恶的Xyz想要抓很多企鹅回去开动物园，当宠物玩。但动物保护协会很快赶来，他必须尽快行动！
　　我们把南极大陆看成一个三维直角坐标系。
　　有N只企鹅，每只企鹅会在一定的时刻的出现，第i只企鹅在Ai时刻出现在坐标为(Bi,Ci,Di)的地方。
　　Xyz要在某一时刻在某一地方(X,Y,Z)撒一张大网，将(0,0,0)到(X,Y,Z)这个大长方体里的企鹅全都网进去捕捉回家(还没出现的企鹅就不会被捉进去了)。
　　为了快准狠而且保证不铺张浪费网，Xyz想知道不同时间不同地点撒网能抓到几个企鹅(这样的询问有Q个)。然后他再行动。

输入

第一行一个整数N表示企鹅个数。
　　第二行到N+1行每行四个实数(Ai,Bi,Ci,Di)，表示企鹅的出现时间和位置
　　第N+2行一个整数Q表示询问个数。
　　接下来Q行每行四个实数(T,X,Y,Z)，表示询问的时间和位置。

输出

输出共Q行，每行一个整数，回答每个询问能抓到几个企鹅。

样例输入

样例输出

1
0

提示

【数据规模和约定】

共20个数据
　　数据1~3 N,Q<=1000
　　数据4~6 N,Q<=5000
　　数据7~10 N,Q<=10000
　　数据11~14 N<=30000,Q<=10000
　　数据15~18 N<=10000,Q<=30000
　　数据19~20 N,Q<=30000

solution

四维偏序，排序+cdq+cdq+树状数组

排序好暴力

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#define maxn 60005
using namespace std;
int n,q,ans[maxn],co,tree[maxn];
map<double,int>ls;
struct node{
	double x,y,z,t;
	int id,fl;
	void bug()
	{
		cout<<t<<' '<<x<<' '<<y<<' '<<z<<' '<<fl<<endl;
	}
}s[maxn],h[maxn];
bool T(node a,node b){
	if(a.t==b.t){
		return a.x<b.x||a.x==b.x&&a.y<b.y||a.x==b.x&&a.y==b.y&&a.z<b.z;
	}
	return a.t<b.t; 
}
bool Z(node a,node b){
	return a.z<b.z;
}
bool Y(node a,node b){
	return a.y<b.y||a.y==b.y&&Z(a,b);
}
bool X(node a,node b){
	return a.x<b.x||a.x==b.x&&Y(a,b);
}

void jia(int k,int v){
//puts("haha");cout<<k<<' '<<co<<endl;
	for(int i=k;i<=co;i+=i&-i)tree[i]+=v;
	
}
int ask(int k){
	int sum=0;
	for(int i=k;i;i-=i&-i)sum+=tree[i];
	return sum;
}
void cd(int l,int r){
	
	if(l==r)return;
	int mid=l+r>>1;
	cd(l,mid);cd(mid+1,r);
	//cout<<"aaa "<<l<<' '<<r<<endl;
	sort(h+l,h+mid+1,Y);sort(h+mid+1,h+r+1,Y);
	
	//for(int i=l;i<=r;i++){cout<<"aa ";h[i].bug();}
	//cout<<endl;
	int i=l;
	for(int j=mid+1;j<=r;j++){
		while(h[i].y<=h[j].y&&i<=mid){
			if(!h[i].fl&&h[i].id==0)jia(ls[h[i].z],1);
			i++;
		}
		if(h[j].fl==1&&h[j].id>0){
			
			ans[h[j].id]+=ask(ls[h[j].z]);
			//cout<<"bbb ";h[i].bug();
			//	cout<<ls[h[i].z]<<endl;
		}
	}
	for(int j=l;j<i;j++)if(!h[j].fl&&h[j].id==0)jia(ls[h[j].z],-1);
}
void cdq(int l,int r){

	if(l==r)return;
	int mid=l+r>>1;
	cdq(l,mid),cdq(mid+1,r);
	sort(s+l,s+mid+1,X);sort(s+mid+1,s+r+1,X);
	int i=l,tot=l-1;
	for(int j=mid+1;j<=r;j++){
		while(s[i].x<=s[j].x&&i<=mid)h[++tot]=s[i++],h[tot].fl=0;
		h[++tot]=s[j],h[tot].fl=1;
	}
	// maybe tot!=r  
	cd(l,tot);
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)scanf("%lf%lf%lf%lf",&s[i].t,&s[i].x,&s[i].y,&s[i].z);
	cin>>q;
	for(int i=n+1;i<=n+q;i++){
		scanf("%lf%lf%lf%lf",&s[i].t,&s[i].x,&s[i].y,&s[i].z);
		s[i].id=i-n;
	}
	int nn=n+q;
	sort(s+1,s+nn+1,Z);
	for(int i=1;i<=nn;i++){
		if(!ls[s[i].z])ls[s[i].z]=++co;
		//cout<<s[i].z<<' '<<ls[s[i].z]<<endl;
	}
	sort(s+1,s+nn+1,T);
    cdq(1,nn);
    for(int i=1;i<=q;i++)printf("%d
",ans[i]);
	return 0;
}