BZOJ 4548 小奇的糖果

Description

有 N 个彩色糖果在平面上。小奇想在平面上取一条水平的线段,并拾起它上方或下方的所有糖果。求出最多能够拾

起多少糖果,使得获得的糖果并不包含所有的颜色。

Input

包含多组测试数据,第一行输入一个正整数 T 表示测试数据组数。

接下来 T 组测试数据,对于每组测试数据,第一行输入两个正整数 N、K,分别表示点数和颜色数。

接下来 N 行,每行描述一个点,前两个数 x, y (|x|, |y| ≤ 2^30 - 1) 描述点的位置,最后一个数 z (1 ≤ z ≤

 k) 描述点的颜色。

对于 100% 的数据,N ≤ 100000,K ≤ 100000,T ≤ 3

Output

对于每组数据在一行内输出一个非负整数 ans,表示答案

Sample Input

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

Sample Output

5

HINT

Source

By Hzwer

~~~~~~~~~~~~~~~~~~~~~~~~~~~~

树状数组+链表~

注意xr[i]要先设置为n+1！

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

int t,n,k,ans,x[100002],y[100001],len1,len2,c[100002],fi[100002],xl[100002],xr[100002];

struct node{
	int x,y,col,id;
}a[100001];

bool operator < (node u,node v)
{
	return u.x<v.x;
}

bool cmp(node u,node v)
{
	return u.y<v.y;
}

int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}

void add(int u,int v)
{
	for(;u<=n+1;u+=u&(-u)) c[u]+=v; 
}

int cal(int u)
{
	int now=0;
	for(;u;u-=u&(-u)) now+=c[u];
	return now;
}

void update(int l,int r)
{
	if(l>r) return;
	ans=max(ans,cal(r)-cal(l-1));
}

void solve()
{
	x[0]=0;x[n+1]=n+1;
	memset(c,0,sizeof(c));
	memset(fi,0,sizeof(fi));
	sort(a+1,a+n+1);
	for(int i=1;i<=n;i++) add(a[i].x,1);
	for(int i=1;i<=n;i++)
	{
		int t=a[i].id,l=fi[a[i].col];
		xl[t]=l;xr[t]=n+1;
		if(l) xr[l]=t;
		update(x[l]+1,x[t]-1);
		fi[a[i].col]=t;
	}
	for(int i=1;i<=k;i++) update(x[fi[i]]+1,n+1);
	sort(a+1,a+n+1,cmp);
	for(int i=1,j=1;i<=n;i++)
	{
		while(j<=n && a[i].y==a[j].y) add(a[j].x,-1),j++;
		xl[xr[a[i].id]]=xl[a[i].id];
		xr[xl[a[i].id]]=xr[a[i].id];
		update(x[xl[a[i].id]]+1,x[xr[a[i].id]]-1);
	}
}

int main()
{
	t=read();
	while(t--)
	{
		n=read();k=read();ans=0;
		for(int i=1;i<=n;i++) x[i]=a[i].x=read(),y[i]=a[i].y=read(),a[i].col=read(),a[i].id=i;
		sort(x+1,x+n+1);sort(y+1,y+n+1);
		len1=unique(x+1,x+n+1)-x-1;len2=unique(y+1,y+n+1)-y-1;
		for(int i=1;i<=n;i++)
		{
			a[i].x=lower_bound(x+1,x+len1+1,a[i].x)-x;
			a[i].y=lower_bound(y+1,y+len2+1,a[i].y)-y;
		}
		for(int i=1;i<=n;i++) x[i]=a[i].x;
		solve();
		for(int i=1;i<=n;i++) a[i].y*=-1;
		solve();
		printf("%d\n",ans);
	}
	return 0;
}