[BZOJ4548]小奇的糖果（双向链表+树状数组）

题目：

我是超链接

题解：

首先，按y坐标排序，我们假定这条线无限低，这时候答案相当于相邻两个同种颜色的中间的最大点数
然后这条线向上，一次性删除一行
用双向链表维护一个点的前一个和后一个同种颜色的位置
删除的话，把这个点从树状数组和链表中都删除，每次删除，记录一下这个点pre和nxt之间的答案
然后把y全都赋成相反数，再做一遍即可得到下端的情况

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=100005;
struct hh{int x,y,c,id;}a[N];
int nxt[N],pre[N],co[N];
int c[N],n,k,b[N],up[N],ans,s,last[N];
void add(int loc,int vv){for (int i=loc;i<=s+1;i+=i&(-i)) c[i]+=vv;}
int qurry(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i)) ans+=c[i];
    return ans;
}
int getans(int l,int r)
{
    if (l>r) return 0;
    return qurry(r)-qurry(l-1);
}
int cmpx(hh a,hh b){return a.x<b.x;}
int cmpy(hh a,hh b){return a.y<b.y;}
void work()
{
    co[n+1]=s+1;/////
    memset(up,0,sizeof(up)); 
    for (int i=1;i<=n;i++) pre[i]=up[a[i].c],up[a[i].c]=i,last[a[i].c]=a[i].x,co[i]=a[i].x;
    memset(up,0,sizeof(up));
    for (int i=n;i>=1;i--) nxt[i]=(up[a[i].c]?up[a[i].c]:n+1),up[a[i].c]=i;
    memset(c,0,sizeof(c));
    for (int i=1;i<=n;i++) 
      ans=max(ans,getans(a[pre[i]].x+1,a[i].x-1)),add(a[i].x,1);
    for (int i=1;i<=k;i++) ans=max(ans,getans(last[i]+1,s));
    sort(a+1,a+n+1,cmpy);
    for (int t=1,i=1;i<=n;i++)
    {
        int now=a[i].id;
        while (t<=n && a[i].y==a[t].y) add(a[t].x,-1),t++;
        if (pre[now]) nxt[pre[now]]=nxt[now];
        if (nxt[now]) pre[nxt[now]]=pre[now];

        ans=max(ans,getans(co[pre[now]]+1,co[nxt[now]]-1));
        pre[now]=nxt[now]=0;
    }
}
int main()
{
    freopen("xin.in","r",stdin);
    int T;scanf("%d",&T);
    while (T--)
    {
        ans=0;
        scanf("%d%d",&n,&k);
        for (int i=1;i<=n;i++) scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].c);
        sort(a+1,a+n+1,cmpx);for (int i=1;i<=n;i++) a[i].id=i;
        for (int i=1;i<=n;i++) b[i]=a[i].x;
        s=unique(b+1,b+n+1)-b-1;
        for (int i=1;i<=n;i++) a[i].x=lower_bound(b+1,b+s+1,a[i].x)-b;
        work();
        for (int i=1;i<=n;i++) a[i].y=-a[i].y;
        sort(a+1,a+n+1,cmpx);
        work();
        printf("%d\n",ans);
    }
}