这道题的本质是:给出点在立方体内的分布,询问一个在给定中心、中心到内边界距离为a、中心到外边界距离为b的空心立方体内的点的个数。
一开始想到多维树状数组之类的。。。。然后发现想多了。。。。预处理以下最后一维的前缀和就好(维护高维也行,但这道题时间充裕,就没必要搞复杂了)
注意处理边界情况。
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
int f[45][45][45];
int s[45][45][45];
void work(int n)
{
int i,j,k,t,m,a,b,q;
int h1,h2,h3,num,ans;
int l1,r1,l2,r2,l3,r3,ll,rr;
memset(f,0,sizeof(f));
memset(s,0,sizeof(s));
while(n--)
{
scanf("%d%d%d%d",&h1,&h2,&h3,&num);
f[h1][h2][h3]+=num;
}
for(i=1;i<=40;i++)
for(j=1;j<=40;j++)
for(k=1;k<=40;k++)
s[i][j][k]=s[i][j][k-1]+f[i][j][k];
/*
for(i=1;i<=3;i++)
{
for(j=1;j<=3;j++)
{
for(k=1;k<=3;k++) printf("s[%d][%d][%d]=%d\n",i,j,k,s[i][j][k]);
printf("\n");
}
printf("\n");
}
*/
scanf("%d",&q);
while(q--)
{
scanf("%d%d%d%d%d",&h1,&h2,&h3,&a,&b);
ans=0;
l1=max(1,h1-b); r1=min(40,h1+b);
for(i=l1;i<=r1;i++)
{
l2=max(1,h2-b); r2=min(40,h2+b);
for(j=l2;j<=r2;j++)
{
l3=max(1,h3-b); r3=min(40,h3+b);
ans+=s[i][j][r3]-s[i][j][l3-1];
}
}
//printf("temp ans=%d\n",ans);
l1=max(1,h1-a+1); r1=min(40,h1+a-1);
//printf("l1=%d r1=%d\n",l1,r1);
for(i=l1;i<=r1;i++)
{
l2=max(1,h2-a+1); r2=min(40,h2+a-1);
//printf("l2=%d r2=%d\n",l2,r2);
for(j=l2;j<=r2;j++)
{
l3=max(1,h3-a+1); r3=min(40,h3+a-1);
ans-=s[i][j][r3]-s[i][j][l3-1];
}
}
printf("%d\n",ans);
}
}
int main()
{
int n;
while(~scanf("%d",&n))
work(n);
return 0;
}