题目
ABC生日收到N支蜡笔,每支蜡笔的颜色是三原色红绿蓝的组合,第i个蜡笔的颜色用Ri表示红色,Gi表示绿色,Bi表示蓝色。
蜡笔i和蜡笔j的颜色差异定义为max(|Ri-Rj|,|Gi-Gj|,|Bi-Bj|),多支蜡笔的颜色差异定义为其中任意两个蜡笔颜色的最大差异值。
ABC想从N支蜡笔中选出K支出来,要求这K支蜡笔的颜色差异值最小。
这题一看到差异的表示形式,对平面图敏感的人(例如我,很多题都莫名奇妙的想到平面图…)就想着是三维的切比雪夫距离..(其实有没有这个东西还是不知道),于是就坑了..
好了,言归正传,这题其实并不用考虑那么麻烦。
显然,我们可以通过二分答案(暴力有点慢,通常都会想到用二分来优化一下),然后判断答案是否合法来收缩答案的范围,最后得到答案。
那么怎么判断答案的合法范围呢?现在我们得知答案ans,那么我们强制颜色差异最大为ans,只要检验合法的最大点集内的点数大于等于k就行了,这个怎么判呢?
暴力枚举R、G、B的下界,然后上界就是R+k、G+k、B+k,这时,我们可以停下来想想,这像什么?以顶点为(R,G,B),边长为k的正方形,我们要求里面有多少个给定的点,二维的情况我们都知道用二维前缀和减去边界部分,再补回减重的,三维的同样,拓展到N维的就类似于容斥原理了(其实表达式的确有点像)..
好在我们打完暴力后,我们打开计算器算一下时间复杂度:
时间复杂度:
O(2553logN)
敲完计算器后,再看看时间限制→5000ms,咱们惊呆了。
贴代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#define N 257
#include<cmath>
#include<ctime>
#define random(x)(rand()%x)
using namespace std;
int n,k,ans,m;
int f[N][N][N];
void init(){
static int x,y,z;
scanf("%d %d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d %d %d",&x,&y,&z),f[x+1][y+1][z+1]++;
}
void pre(){
for (int i=1;i<N;i++)
for (int j=1;j<N;j++)
for (int k=1;k<N;k++){
f[i][j][k]+=f[i-1][j][k];
f[i][j][k]+=f[i][j-1][k];
f[i][j][k]+=f[i][j][k-1];
f[i][j][k]-=f[i-1][j-1][k];
f[i][j][k]-=f[i][j-1][k-1];
f[i][j][k]-=f[i-1][j][k-1];
f[i][j][k]+=f[i-1][j-1][k-1];
}
}
bool jian(int x){
static int xx,yy,zz;
for (int i=1;i<N;i++)
for (int j=1;j<N;j++)
for (int k=1;k<N;k++){
xx=min(i+x,N-1);
yy=min(j+x,N-1);
zz=min(k+x,N-1);
if (f[xx][yy][zz]-f[i-1][yy][zz]-f[xx][j-1][zz]-f[xx][yy][k-1]+f[i-1][j-1][zz]+f[i-1][yy][k-1]+f[xx][j-1][k-1]-f[i-1][j-1][k-1]>=m)return 1;
}
return 0;
}
void work(){
static int l,r,mid;
l=0,r=255;
while (l<=r)
if (jian(mid=(l+r)/2))r=mid-1;else l=mid+1;
if (l<0)l=0;
while (!jian(l))l++;
while (l>0&&jian(l-1))l--;
ans=l;
}
void write(){
printf("%d",ans);
}
int main(){
srand((int)time(0));
init();
pre();
work();
write();
return 0;
}