小奇的糖果(candy)

【题目背景】
小奇不小心让糖果散落到了地上，它对着满地的彩色糖果胡思乱想。
【问题描述】
有 N 个彩色糖果在平面上。 小奇想在平面上取一条水平的线段,并拾起它上方或
下方的所有糖果。求出最多能够拾起多少糖果,使得获得的糖果并不包含所有的
颜色。
【输入格式】
包含多组测试数据,第一行输入一个正整数 T 表示测试数据组数。
接下来 T 组测试数据,对于每组测试数据,第一行输入两个正整数 N、K,分别表
示点数和颜色数。
接下来 N 行,每行描述一个点,前两个数 x, y (|x|, |y| ≤ 2^30 - 1) 描述点
的位置,最后一个数 z (1 ≤ z ≤ k) 描述点的颜色。
【输出格式】
对于每组数据在一行内输出一个非负整数 ans,表示答案。
【样例输入】
1
10 3
1 2 3
2 1 1
2 4 2
3 5 3
4 4 2
5 1 2
6 3 1
6 7 1
7 2 3
9 4 2
【样例输出】
5
【数据范围】
对于 30% 的数据,N ≤ 100;
对于 60% 的数据,N ≤ 5000;
对于 100% 的数据,N ≤ 100000,K ≤ 100000,T ≤ 3

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<algorithm>
 #include<cstring>
 #include<queue>
 #include<vector>
 #include<set>
 using namespace std;
 typedef long long LL;
 const int maxn=;
 int T,ANS,N,K;
 int last[maxn],l[maxn],r[maxn];
 int disc[maxn],x[maxn],b[maxn];
 struct Candy{
     int x,y,k;
     int id;
 }a[maxn];
 inline int cmpbyy(const Candy & w,const Candy & e){//´Óϵ½ÉÏ ´Ó×óµ½ÓÒ
     return w.y<e.y||(w.y==e.y&&w.x<e.x);
 }
 inline int cmpbyx(const Candy & w,const Candy & e){//´Ó×óµ½ÓÒ ´Óϵ½ÉÏ
     return w.x<e.x||(w.x==e.x&&w.y>e.y);
 }
 struct Tree{
     int l,r,sum;
 }tr[maxn*];
 inline void Build(int rt,int l,int r){
     tr[rt].l=l; tr[rt].r=r;
     if(l==r){
         tr[rt].sum=b[l];
         return ;
     }
     int mid=(l+r)>>;
     Build(rt<<,l,mid); Build(rt<<|,mid+,r);
     tr[rt].sum=tr[rt<<].sum+tr[rt<<|].sum;
 }
 inline void change(int rt,int pos,int delta){
     if(tr[rt].l==tr[rt].r){
         tr[rt].sum+=delta;
         return ;
     }
     int mid=(tr[rt].l+tr[rt].r)>>;
     if(pos<=mid) change(rt<<,pos,delta);
     else change(rt<<|,pos,delta);
     tr[rt].sum=tr[rt<<].sum+tr[rt<<|].sum;
 }
 inline int query(int rt,int l,int r){
     if(l<=tr[rt].l&&tr[rt].r<=r){
         return tr[rt].sum;
     }
     int mid=(tr[rt].l+tr[rt].r)>>;
     int ans=;
     if(l<=mid) ans+=query(rt<<,l,r);
     if(mid+<=r) ans+=query(rt<<|,l,r);
     return ans;
 }
 inline void update(int l,int r){
     if(l>r) return ;
     ANS=max(ANS,query(,l,r));
 }
 inline void work(){
     x[]=; x[N+]=N+;
     memset(last,,sizeof(last)); memset(b,,sizeof(b));
     for(int i=;i<=;i++) tr[i].l=tr[i].r=tr[i].sum=;

     sort(a+,a+N+,cmpbyx);
     for(int i=;i<=N;i++) b[x[i]]++;
     Build(,,N+);
     for(int i=;i<=N;i++){
         int tmp=a[i].id,L=last[a[i].k];
         l[tmp]=L; r[tmp]=N+;
         if(L) r[L]=tmp;
         update(x[L]+,x[tmp]-);
         last[a[i].k]=tmp;
     }
     for(int i=;i<=K;i++){
         update(x[last[i]]+,N+);
     }
     sort(a+,a+N+,cmpbyy);
     for(int i=,j=;i<=N;i++){
         int tmp=a[i].id;
         while(j<=N&&a[j].y==a[i].y){
             change(,a[j].x,-);
             j++;
         }
         l[r[tmp]]=l[tmp]; r[l[tmp]]=r[tmp];
         update(x[l[tmp]]+,x[r[tmp]]-);
     }
 }
 int main(){
 //    freopen("candy.in","r",stdin);
 //    freopen("candy.out","w",stdout);
     scanf("%d",&T);
     while(T--){
         ANS=;
         scanf("%d%d",&N,&K);
         for(int i=;i<=N;i++){
             scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].k);
             a[i].id=i;
         }
         for(int i=;i<=N;i++){
             disc[i]=a[i].x;
         }
         sort(disc+,disc+N+);
         int *end=unique(disc+,disc+N+);
         for(int i=;i<=N;i++){
             a[i].x=lower_bound(disc+,end,a[i].x)-disc;
             x[i]=a[i].x;
         }
         work();
         for(int i=;i<=N;i++) a[i].y*=-;
         work();
         printf("%d\n",ANS);
     }
     return ;
 }