GDOI2016模擬8.18蠟筆
阿新 • • 發佈:2019-02-14
題目
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維的就類似於容斥原理了(其實表示式的確有點像)..
好在我們打完暴力後,我們開啟計算器算一下時間複雜度:
時間複雜度:
敲完計算器後,再看看時間限制→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;
}