1. 程式人生 > >GDOI2016模擬8.18蠟筆

GDOI2016模擬8.18蠟筆

題目
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維的就類似於容斥原理了(其實表示式的確有點像)..

好在我們打完暴力後,我們開啟計算器算一下時間複雜度:
時間複雜度:O2553logN

敲完計算器後,再看看時間限制→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; }