平面上有n个点,每个点有k种颜色中的一个。
你可以选择一条水平的线段获得在其上方或其下方的所有点。
请求出你最多能够得到多少点,使得获得的点并不包含所有的颜色。
我好弱啊。
发现我们如果枚举上边界。
那么我们的答案区间一定在上边界下的两个相邻的同色点(或左边界和某色点,或某色点和右边界)之间。
发现上边界从上往下的过程中只会删除。
那么使用双向链表
O
(
1
)
O(1)
O(1)维护相邻的同色点。
树状数组
O
(
log
n
)
O(\log n)
O(logn)维护相邻的同色点中包含的点数。
边删边求答案。
AC Code:
#include<bits/stdc++.h>
#define maxn 100005
using namespace std;
int n,k,x[maxn],y[maxn],z[maxn],sb[maxn],c[maxn],cx[maxn];
int ans = 0;
int pre[maxn] , suc[maxn] , loc[maxn];
bool cmp(const int &u,const int &v){ return y[u] < y[v]; }
bool cmp2(const int &u,const int &v){ return x[u] < x[v]; }
int tr[maxn];
void upd(int now,int v){ for(;now<=n;now+=now&-now) tr[now]+=v;}
int qsum(int now){ int ret = 0;for(;now;now-=now&-now) ret+=tr[now];return ret; }
void solve(){
sort(c+1,c+1+n,cmp);
memset(loc,0,sizeof loc);
memset(tr,0,sizeof tr);
memset(pre,0,sizeof pre);
memset(suc,0,sizeof suc);
for(int i=1;i<=n;i++) upd(x[i],1);
for(int i=1;i<=n;i++){
int u = cx[i];
pre[u] = loc[z[u]] , suc[loc[z[u]]] = u;
ans = max(ans , qsum(x[u]-1)-qsum(x[pre[u]]));
loc[z[u]] = u;
}
x[n+1] = sb[0] + 1;
for(int i=1;i<=k;i++)
pre[n+1] = loc[i] , suc[loc[i]] = n+1,
ans = max(ans , qsum(n) - qsum(x[loc[i]]));
for(int i=1,j;i<=n;i=j){
for(j=i;j<=n && y[c[j]] == y[c[i]];j++)
upd(x[c[j]],-1);
for(j=i;j<=n && y[c[j]] == y[c[i]];j++){
int u = c[j];
ans = max(ans , qsum(x[suc[u]]-1) - qsum(x[pre[u]]));
pre[suc[u]] = pre[u];
suc[pre[u]] = suc[u];
}
}
}
int main(){
int T;
for(scanf("%d",&T);T--;){
sb[0] = ans = 0;
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++){
scanf("%d%d%d",&x[i],&y[i],&z[i]);
sb[++sb[0]] = x[i];
c[i] = i , cx[i] = i;
}
sort(sb+1,sb+1+sb[0]);
sb[0] = unique(sb+1,sb+1+sb[0])-sb-1;
for(int i=1;i<=n;i++) x[i] = lower_bound(sb+1,sb+1+sb[0],x[i])-sb;
sort(cx+1,cx+1+n,cmp2);
solve();
for(int i=1;i<=n;i++) y[i] = -y[i];
solve();
printf("%d\n",ans);
}
}