[Jzoj] 3186. 蜡笔

题目大意

$ABC$生日收到 $N$支蜡笔，每支蜡笔的颜色是三原色红绿蓝的组合，第 $i$个蜡笔的颜色用 $R_i$表示红色， $G_i$表示绿色， $B_i$表示蓝色。

蜡笔 $i$和蜡笔 $j$的颜色差异定义为 $max(|Ri-Rj|,|Gi-Gj|,|Bi-Bj|)$，多支蜡笔的颜色差异定义为其中任意两个蜡笔颜色的最大差异值。

$ABC$想从 $N$支蜡笔中选出 $K$支出来，要求这 $K$支蜡笔的颜色差异值最小。

题目解析

我们把颜色看作为空间上的点

因为输入坐标的值很小，所以利用前缀和，检查其区间是否至少有K点

暴力枚举即可

代码

#include<bits/stdc++.h>
#define N 257
using namespace std;
int n,k,ans,m;
int f[N][N][N];
bool check(int x)
{
    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;
}
int main()
{
    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]++;
    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];
      }
    int l=0,r=255,mid;
    while (l<=r)
     if(check(mid=(l+r)/2))r=mid-1,ans=mid;else l=mid+1;
    printf("%d",ans);
    return 0;
}