BZOJ3658 : Jabberwocky

考虑将某线段下方的点取走：

将所有点从低到高排序

每扫描到一条水平线，对于上面每个点，找到它下面同色的前驱后继，统计中间点的个数

然后再把线上所有点插入数据结构中

最后再统计相邻的同色的点之间的点个数

用动态开点的权值线段树+树状数组维护，时间复杂度$O(n\log n)$。

考虑将某线段上方的点取走：

把扫描线的顺序反过来即可

注意特判出现颜色数没达到k的情况。

 

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100010,M=1800000;
int C,n,k,i,j,ans,bit[N],b[N],vis[N];
struct P{int x,y,c;}a[N];
inline bool cmpc(P a,P b){return a.c==b.c?a.x<b.x:a.c<b.c;}
inline bool cmpy(P a,P b){return a.y<b.y;}
inline void up(int x){if(ans<x)ans=x;}
inline int lower(int x){
  int l=1,r=n,t,mid;
  while(l<=r)if(b[mid=(l+r)>>1]<=x)l=(t=mid)+1;else r=mid-1;
  return t;
}
inline void add(int x){for(;x<=n;x+=x&-x)bit[x]++;}
inline int ask(int x){int t=0;for(;x;x-=x&-x)t+=bit[x];return t;}
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
int T[N],vl[M],vr[M],l[M],r[M],tot,pre,nxt;
void ins(int&x,int a,int b,int c){
  if(!x)x=++tot;
  if(a==b){vl[x]=vr[x]=a;return;}
  int mid=(a+b)>>1;
  if(c<=mid)ins(l[x],a,mid,c);else ins(r[x],mid+1,b,c);
  vl[x]=vl[l[x]]?vl[l[x]]:vl[r[x]];
  vr[x]=vr[r[x]]?vr[r[x]]:vr[l[x]];
}
void getl(int x,int a,int b,int c){
  if(!x||nxt)return;
  if(c<=a){nxt=vl[x];return;}
  int mid=(a+b)>>1;
  if(c<=mid)getl(l[x],a,mid,c);
  getl(r[x],mid+1,b,c);
}
void getr(int x,int a,int b,int d){
  if(!x||pre)return;
  if(b<=d){pre=vr[x];return;}
  int mid=(a+b)>>1;
  if(d>mid)getr(r[x],mid+1,b,d);
  getr(l[x],a,mid,d);
}
int main(){
  for(read(C);C--;printf("%d\n",ans)){
    read(n),read(k);
    for(i=1;i<=k;i++)vis[i]=0;
    for(ans=0,i=1;i<=n;i++)read(a[i].x),read(a[i].y),read(a[i].c),b[i]=a[i].x,vis[a[i].c]=1;
    for(sort(b+1,b+n+1),sort(a+1,a+n+1,cmpy),i=1;i<=n;i++)a[i].x=lower(a[i].x);
    for(i=1;i<=n;i++)bit[i]=0;
    for(i=1;i<=k;i++)if(!vis[i]){ans=n;break;}
    if(ans)continue;
    for(i=1;i<=tot;i++)vl[i]=vr[i]=l[i]=r[i]=0;tot=0;
    for(i=1;i<=k;i++)T[i]=0,ins(T[i],0,n+1,0),ins(T[i],0,n+1,n+1);
    for(i=1;i<=n;i=j){
      for(j=i;j<=n&&a[j].y==a[i].y;j++){
        pre=nxt=0,getl(T[a[j].c],0,n+1,a[j].x),getr(T[a[j].c],0,n+1,a[j].x);
        up(ask(nxt-1)-ask(pre));
      }
      for(j=i;j<=n&&a[j].y==a[i].y;j++)ins(T[a[j].c],0,n+1,a[j].x),add(a[j].x);
    }
    for(i=1;i<=n;i++)bit[i]=0;
    for(i=1;i<=tot;i++)vl[i]=vr[i]=l[i]=r[i]=0;tot=0;
    for(i=1;i<=k;i++)T[i]=0,ins(T[i],0,n+1,0),ins(T[i],0,n+1,n+1);
    for(i=n;i;i=j){
      for(j=i;j&&a[j].y==a[i].y;j--){
        pre=nxt=0,getl(T[a[j].c],0,n+1,a[j].x),getr(T[a[j].c],0,n+1,a[j].x);
        up(ask(nxt-1)-ask(pre));
      }
      for(j=i;j&&a[j].y==a[i].y;j--)ins(T[a[j].c],0,n+1,a[j].x),add(a[j].x);
    }
    for(sort(a+1,a+n+1,cmpc),i=1;i<=n;i++)bit[i]=0;
    for(i=1;i<=n;i++)bit[a[i].x]++;
    for(i=2;i<=n;i++)bit[i]+=bit[i-1];
    for(i=1;i<n;i++)if(a[i].c==a[i+1].c)up(bit[a[i+1].x-1]-bit[a[i].x]);
    else up(max(n-bit[a[i].x],bit[a[i+1].x-1]));
    up(max(bit[a[1].x-1],n-bit[a[n].x]));
  }
  return 0;
}