【题解】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;
}