玲珑杯1143 - 计算几何你瞎暴力怕【思维】

DESCRIPTION

今天$HH$考完了期末考试,他在教学楼里闲逛,他看着教学楼里一间间的教室,于是开始思考:

如果从一个坐标为 $(x_1,y_1,z_1)$的教室走到$(x_2,y_2,z_2)$的距离为 $|x_1-x_2|+|y_1-y_2|+|z_1-z_2|$

那么有多少对教室之间的距离是不超过$R$的呢?

INPUT

第一行是一个整数 $T(1\le T\le 10)$, 表示有 $T$组数据接下来是 $T$组数据,对于每组数据:第一行是两个整数 $n,q(1\le n\le 5\times {10}^4,1\le q\le {10}^3)$, 表示有 $n$间教室, $q$次询问.接下来是 $n$行, 每行3个整数 $x_i,y_i,z_i(0\le x_i,y_i,z_i\le 10)$,表示这间教室的坐标.最后是 $q$行,每行一个整数 $R(0\le R\le {10}^9)$,意思见描述.

OUTPUT

对于每个询问 $R$输出一行一个整数,表示有多少对教室满足题目所述的距离关系.

SAMPLE INPUT

13 30 0 01 1 11 1 1123

SAMPLE OUTPUT

113

HINT

对于样例,1号教室和2号教室之间的距离为3, 1号和3号之间的距离为3, 2号和3号之间的距离为0

SOLUTION

“玲珑杯”ACM比赛 Round #18

思路：n*n 肯定过不了，坐标比较小，可枚举，注意注释bug点；

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define max_n 2000010
using namespace std;
typedef long long LL;

LL a[max_n],b[100],c[100];

struct node
{
	int x;
	int y;
	int z;
}s[2000];

int main()
{
	int t,p1,p2,p3;
	LL n,q;
	scanf("%d",&t);
	while(t--)
	{
		int j=0;
		memset(s,0,sizeof(s));
		memset(a,0,sizeof(a));
		memset(b,0,sizeof(b));
		memset(c,0,sizeof(c));
		scanf("%lld %lld",&n,&q);
		for(int i=0;i<n;i++)
		{
			scanf("%d %d %d",&p1,&p2,&p3);
			int ans=p1*11+p2*200+p3*3000;
			if(a[ans]==0)
			{
				s[j].x=p1,s[j].y=p2,s[j].z=p3;
				j++;
			}
			a[ans]++;
		}
		for(int i=0;i<j;i++)
		{
			for(int p=i;p<j;p++)
			{
				int res1=s[i].x*11+s[i].y*200+s[i].z*3000;
				int res2=s[p].x*11+s[p].y*200+s[p].z*3000;
				if(i!=p)
				{
					int ans=abs(s[i].x-s[p].x)+abs(s[i].y-s[p].y)+abs(s[i].z-s[p].z);
					b[ans]+=a[res1]*a[res2];
				}
				else
				{
					if(a[res1])//注意这个bug点； 
					{
						b[0]+=a[res1]*(a[res1]-1)/2;
					}
				}
			}
		}
		LL cnt=0,r;
		for(int i=0;i<=35;i++)
		{
			cnt+=b[i];
			c[i]=cnt;
		}
		while(q--)
		{
			scanf("%lld",&r);
			if(r>=32) printf("%lld\n",c[32]);
			else printf("%lld\n",c[r]);
		}
	} 
	return 0;
}