【题解】BZOJ4548 小奇的糖果(树状数组)

【题解】BZOJ4548 小奇的糖果(树状数组)

说在前面:我有个同学叫小奇,他有一个朋友叫达达,达达特爱地理和旅游,初中经常AK地理,好怀恋和他已经达达一起到当时初中附近许多楼盘的顶楼逛的时光...

主要是今天大家讲题的时候我偷偷溜出来到了科技楼七楼,从窗户爬到阳台上,发现顶楼居然有:

  • 天文望远镜一台
  • 气象观测室一间
  • 化学实验室一间
  • 水文实验室一间
  • Corridors一个
  • 奇怪的植物若干
  • 没有监控

考虑到要求至少没有一种颜色,那么就钦定某种颜色不存在即可

考虑这样一种做法,将所有点按照y从小往大排序,枚举到\(u=(x,y,z)\)时,查询y比他小的\(z\)颜色的点的位置,定位到在\(u\)两侧的点然后查询在\(u\)以下的点的个数,此时由于一定没有\(z\)颜色,所以是合法的。

然后那些一条线往上的方案,可以把y*=-1解决

//@winlere
#include
#include
#include
#include
#include
#include
#define DEBUG(s) cerr<<(#s)<<" = "<<(s)< s[maxn];
int seg[maxn];

inline void add(const int&pos,const int&tag){
    for(int t=pos;t<=len;t+=t&-t) seg[t]+=tag;
}

inline int que(const int&pos){
    int ret=0;
    for(int t=pos;t>0;t-=t&-t) ret+=seg[t];
    return ret;
}

int work(const int&_f){
    int ret=0;
    for(int t=1;t<=n&&_f;++t) data[t].x=qr(),data[t].y=qr(),data[t].z=qr();
    memset(seg,0,sizeof seg);
    for(int t=1;t<=n;++t) sav[t]=data[t].x;
    sav[n+1]=inf; sav[n+2]=-inf;
    sort(sav+1,sav+n+2+1);
    len=unique(sav+1,sav+n+2+1)-sav-1;
    for(int t=1;t<=n;++t) data[t].x=lower_bound(sav+1,sav+len+1,data[t].x)-sav;
    sort(data+1,data+n+1);
    for(int t=1;t<=k;++t) s[t].clear(),s[t].insert(1),s[t].insert(len);
    for(int t=1;t<=n;++t){
        if(s[data[t].z].find(data[t].x)==s[data[t].z].end()){
            ret=max(ret,que(*s[data[t].z].upper_bound(data[t].x)-1)-que(*--s[data[t].z].upper_bound(data[t].x)));
            s[data[t].z].insert(data[t].x);
        }
        add(data[t].x,1);
    }
    for(int t=1;t<=k;++t){
        int last=1;
        for(set::iterator i=s[t].upper_bound(1);i!=s[t].end();++i)
            ret=max(ret,que(*i-1)-que(last)),last=*i;
    }
    if(_f){
        for(int t=1;t<=n;++t) data[t].y=-data[t].y;
        ret=max(ret,work(0));
    }
    return ret;
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("cpp.in","r",stdin);
    freopen("cpp.out","w",stdout);
#endif
    int T=qr();
    while(T--) n=qr(),k=qr(),printf("%d\n",work(1));    
    return 0;
}

你可能感兴趣的:(【题解】BZOJ4548 小奇的糖果(树状数组))