TJU-3852 Haitang1（其他）

来源：互联网发布：算法与复杂性理论编辑：程序博客网时间：2024/06/03 23:38

这道题的本质是：给出点在立方体内的分布，询问一个在给定中心、中心到内边界距离为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;}

0 0