【树状数组+双向链表】bzoi4548 小奇的糖果

传送门

 

先考虑只取上面部分。

假设这个线段的纵坐标无穷小。那么就只看两个相邻的同色点之间的点的个数（不含这两个点）。

先按横坐标排序，然后用树状数组记录一下。

然后把线段慢慢往上移动。（按纵坐标排序）

假设当前点为A，考虑把纵坐标小于等于A纵坐标的点的贡献消去，然后把该点在链表上删去即可。

计算完成后把纵坐标负一下，再做一遍就相当于是只取下面部分。

 

 

几个小细节：

数组清零要在work函数里面。

坐标可能是负的。。读优要把负数考虑进去

#include<bits/stdc++.h>
#define lowbit(x) (x&(-x))
using namespace std;
const int maxn=1e5+10;
int T,N,M,K,x,y,z,ans=0,l[maxn],r[maxn],pre[maxn],b[maxn],tr[maxn];
struct point{int x,y,col,id;}a[maxn];
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
	while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return x*f;
}
inline void change(int x,int v){for(int i=x;i<=M;i+=lowbit(i)) tr[i]+=v;}
inline int query(int x){int ret=0;for(int i=x;i;i-=lowbit(i)) ret+=tr[i];return ret;}
inline bool cmpx(const point &a,const point &b){return a.x<b.x;}
inline bool cmpy(const point &a,const point &b){return a.y<b.y;}
inline void update(int l,int r){if(l<=r)ans=max(ans,query(r)-query(l-1));}
inline void disc(){
    sort(b+1,b+N+1),M=unique(b+1,b+N+1)-(b+1);
    for(int i=1;i<=N;++i)a[i].x=lower_bound(b+1,b+M+1,a[i].x)-b;
    for(int i=1;i<=N;++i)b[i]=a[i].x;
}
inline void init(){
	ans=0,N=read(),K=read();
	for(int i=1;i<=N;++i)
		a[i].x=read(),a[i].y=read(),a[i].col=read(),
		a[i].id=i,b[i]=a[i].x;
}
inline void work(){
	memset(tr,0,sizeof(tr)),memset(pre,0,sizeof(pre));
	sort(a+1,a+N+1,cmpx),b[0]=0,b[N+1]=M+1;
	for(int i=1;i<=N;++i) change(a[i].x,1);
	for(int i=1;i<=N;++i){
		int x=a[i].id,y=pre[a[i].col];
		l[x]=y,r[y]=x,r[x]=N+1,update(b[y]+1,b[x]-1),pre[a[i].col]=x;
	}
	for(int i=1;i<=K;++i) update(b[pre[i]]+1,M);
	sort(a+1,a+N+1,cmpy);
	for(int i=1,j=1;i<=N;++i){
		int x=a[i].id;
		while(j<=N&&a[j].y==a[i].y)change(a[j].x,-1),j++;
		update(b[l[x]]+1,b[r[x]]-1),l[r[x]]=l[x],r[l[x]]=r[x];
	}
}
inline void print(int x){
	if(x>9) print(x/10);
	putchar(x%10+'0');
}
int main(){
	T=read();
	while(T--){
		init(),disc(),work();
		for(int i=1;i<=N;++i) a[i].y=-a[i].y;
		work(),print(ans),putchar('\n');
	}
}