Description
有 N 个彩色糖果在平面上。小奇想在平面上取一条水平的线段,并拾起它上方或下方的所有糖果。求出最多能够拾起多少糖果,使得获得的糖果并不包含所有的颜色。
Input
包含多组测试数据,第一行输入一个正整数 T 表示测试数据组数。
接下来 T 组测试数据,对于每组测试数据,第一行输入两个正整数 N、K,分别表示点数和颜色数。
接下来 N 行,每行描述一个点,前两个数 x, y (|x|, |y| ≤ 2^30 - 1) 描述点的位置,最后一个数 z (1 ≤ z ≤ k) 描述点的颜色。
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
Data Constraint
对于 30% 的数据,N ≤ 100;
对于 60% 的数据,N ≤ 5000;
对于 100% 的数据,N ≤ 100000,K ≤ 100000,T ≤ 3。
Solution
比赛时想着扫描线+线段树什么的,但是没有想出它那个线段怎么处理
正解不麻烦(不像某人打什么主席树)
先只考虑扫描线以上的(以下的类似)
对于每个点,记录它前面和后面最靠近它且颜色相同的点是哪两个,这个可以用双向链表维护,因为扫描线扫过了就要删除
再用一个树状数组维护区间中的点数
那么在我扫描线扫到一条线上时,首先要把线上的点从树状数组中删去
同时维护答案,一个点在删除后最大的区间就是双向链表维护的两边最近颜色相同的点,之间的区间,用树状数组可以找到有多少点,统计答案,随后将这个点从双向链表中删除
还有一点,就是扫描线在最底下的情况,那么直接枚举每个点,统计答案即可
那么扫描线以下的就反过来做一次就行了
我一共打了六个排序(Pascal选手赶快转C++吧)
Code
#include<cstdio>
#include<algorithm>
#include<cstring>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 101000
#define cl(a) memset(a,0,sizeof(a))
using namespace std;
int n,m,up[N],next[N],tot=0,t[N],fst[N],lst[N],ans=0;
struct node{
int x,y,c,i;
}q[N],a[N];
bool cnt1(node x,node y){return x.x<y.x;}
bool cnt2(node x,node y){return x.y<y.y;}
bool cnt3(node x,node y){return x.i<y.i;}
bool cnt4(node x,node y){return x.c<y.c||(x.c==y.c&&x.x<y.x);}
int lowbit(int x){return x&(-x);}
void ins(int x,int y){for(;x<=n;x+=lowbit(x)) t[x]+=y;}
int get(int x)
{
int ans=0;
for(;x;x-=lowbit(x)) ans+=t[x];
return ans;
}
void del(int x)
{
up[next[x]]=up[x];
next[up[x]]=next[x];
up[x]=next[x]=0;
}
void work()
{
int j=1;
fo(i,1,n)
{
int k=j;
while(q[j].y==i) ins(q[j].x,-1),j++;
fo(l,k,j-1)
{
int jy1=next[q[l].i],jy2=up[q[l].i];
ans=max(ans,get(a[jy1].x-1)-get(a[jy2].x));
del(q[l].i);
}
}
}
int main()
{
int ac;scanf("%d",&ac);
for(;ac;ac--)
{
cl(up);cl(next);cl(t);
scanf("%d%d",&n,&m);ans=0;
fo(i,1,n) scanf("%d%d%d",&q[i].x,&q[i].y,&q[i].c),q[i].i=i,a[i]=q[i];
int tot=0;
sort(a+1,a+n+1,cnt1);
fo(i,1,n) if(a[i].x==a[i-1].x) q[a[i].i].x=q[a[i-1].i].x;else q[a[i].i].x=++tot;
sort(a+1,a+n+1,cnt2);tot=0;
fo(i,1,n) if(a[i].y==a[i-1].y) q[a[i].i].y=q[a[i-1].i].y;else q[a[i].i].y=++tot;
sort(a+1,a+n+1,cnt3);
fo(i,1,n) a[i]=q[i],ins(a[i].x,1);//,printf("%d %d %d\n",a[i].x,a[i].y,a[i].c);
sort(q+1,q+n+1,cnt4);
fo(i,1,n)
{
if(q[i].c==q[i-1].c) up[q[i].i]=q[i-1].i,next[q[i-1].i]=q[i].i;
else fst[q[i].c]=q[i].x,lst[q[i-1].c]=q[i-1].x,up[q[i].i]=0,next[q[i-1].i]=n+1;
}
lst[q[n].c]=q[n].x;next[q[n].i]=n+1;
a[n+1].x=n+1;
fo(i,1,m)
{
if(fst[i]==0) {ans=n;break;}
ans=max(ans,get(fst[i]-1));
ans=max(ans,get(n)-get(lst[i]));
}
fo(i,1,n) ans=max(ans,get(a[i].x-1)-get(a[up[i]].x));
sort(q+1,q+n+1,cnt2);
work();
sort(q+1,q+n+1,cnt4);
fo(i,1,n)
{
ins(a[i].x,1),q[i].y=n-q[i].y+1;
if(q[i].c==q[i-1].c) up[q[i].i]=q[i-1].i,next[q[i-1].i]=q[i].i;
else up[q[i].i]=0,next[q[i-1].i]=n+1;
}
next[q[n].i]=n+1;
sort(q+1,q+n+1,cnt2);
work();
printf("%d\n",ans);
}
}