码迷,mamicode.com
首页 > 其他好文 > 详细

acdream1197 Points In Cuboid(hash树状数组)

时间:2014-09-07 02:11:44      阅读:230      评论:0      收藏:0      [点我收藏+]

标签:blog   http   ar   for   div   sp   log   on   amp   

题目链接:http://acdream.info/problem?pid=1197

题意:给出三维空间n个点,m个查询,每次查询某个立方体内的点的个数。

思路:按照一维排序,根据查询插入,其他两位用二位树状数组维护。由于这个坐标太大,二位数组开不出来。这时候就是hash,对于一个位置(x,y),哈希成一个数,作为下标。查询的时候不存在的数字为0.

const int mod=4000007;
const int INF=1000000005;
const int N=100005;
const int M=20011;

struct node
{
	int x,y,z;
};

node a[N],b[N][2];
int n,m;

int ans[N];

pii Q[N];

int cmp(node a,node b)
{
	return a.z<b.z;
}

int A[mod];
int arr[mod];

inline int ha(int x)
{
	int k=x%mod;
	int i;
	for(i=k;;i++)
	{
		if(i==mod) i=0;
		if(A[i])
		{
			if(A[i]==x) return i;
		}
		else
		{
			A[i]=x;
			return i;
		}
	}
}

inline int find(int x)
{
	int k=x%mod;
	int i;
	for(i=k;;i++)
	{
		if(i==mod) i=0;
		if(A[i])
		{
			if(A[i]==x) return i;
		}
		else
		{
			return i;
		}
	}
	return 0;
}

int que(int x,int y)
{
	if(!x||!y) return 0;
	int ans=0;
	while(x)
	{
		int i;
		for(i=y;i;i-=i&-i) 
		{
			ans+=arr[find(x*M+i)];
		}
		x-=x&-x;
	}
	return ans;
}

void add(int x,int y)
{
	while(x<M)
	{
		int i=y;
		while(i<M) arr[ha(x*M+i)]++,i+=i&-i;
		x+=x&-x;
	}
}

int cal(int t)
{
	int ans=0;
	ans+=que(b[t][1].x,b[t][1].y);
	ans-=que(b[t][1].x,b[t][0].y-1);
	ans-=que(b[t][0].x-1,b[t][1].y);
	ans+=que(b[t][0].x-1,b[t][0].y-1);
	return ans;
}

void cal()
{
	clr(A,0);
	clr(arr,0);
	int i;
	for(i=1;i<=n;i++)
	{
		scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
		a[i].x+=10005;
		a[i].y+=10005;
	}
	scanf("%d",&m);
	int j;
	for(i=1;i<=m;i++)
	{
		for(j=0;j<2;j++)
		{
			scanf("%d%d%d",&b[i][j].x,&b[i][j].y,&b[i][j].z);
			b[i][j].x+=10005;
			b[i][j].y+=10005;
		}
		if(b[i][0].x>b[i][1].x) swap(b[i][0].x,b[i][1].x);
		if(b[i][0].y>b[i][1].y) swap(b[i][0].y,b[i][1].y);
		if(b[i][0].z>b[i][1].z) swap(b[i][0].z,b[i][1].z);

		Q[i*2-1]=MP(b[i][0].z-1,-i);
		Q[i*2]=MP(b[i][1].z,i);
	}
	sort(a+1,a+n+1,cmp);
	sort(Q+1,Q+2*m+1);
	j=1;
	for(i=1;i<=m;i++) ans[i]=0;
	for(i=1;i<=n;i++)
	{
		while(j<=2*m&&Q[j].first<a[i].z)
		{
			int x=Q[j].second;
			if(x>0) ans[x]+=cal(x);
			else ans[-x]-=cal(-x);
			j++;
		}
		add(a[i].x,a[i].y);
	}		
	while(j<=2*m)
	{
		int x=Q[j].second;
		if(x>0) ans[x]+=cal(x);
		else ans[-x]-=cal(-x);
		j++;
	}
	for(i=1;i<=m;i++) printf("%d\n",ans[i]);
}

int main()
{
	int num=0;
	while(scanf("%d",&n)!=-1)
	{
		printf("Case #%d:\n",++num);
		cal();
	}
}

 

acdream1197 Points In Cuboid(hash树状数组)

标签:blog   http   ar   for   div   sp   log   on   amp   

原文地址:http://www.cnblogs.com/jianglangcaijin/p/3960034.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!