【GDOI2018模拟7.14】小奇的糖果 树状数组+扫描线

好像上去讲的有点快，主要是本身就挺短，，这里讲的会详细一点。
题意很明显就不说了。

题意就不说了，注意线段上的点是不能选的。
首先要明确一个东西，这里说是双向链表维护，是因为我们维护了一个点的前驱后继，这符合双向链表的定义，但是实际上大家把他当作普通数组一样维护就好了。

先把横坐标离散化，注意是线性离散的话初值不能是0因为有的坐标可以是0（因为这个所以WA0了）。
首先我们假设我们画的那一条直线在无限低的地方，这时候答案相当于相邻两个同种颜色的中间的最大点数。（注意不包括这种颜色）
怎么求？
把横坐标从小到大排序，维护l[i],r[i]分别表示距离i最近的相同颜色的点的编号。
直接维护比较麻烦，设一个last[i]表示i这种颜色最近一次出现的地方在哪个位置，然后用last维护l,r。
维护的同时维护答案，用树状数组和l,r维护一下区间和就好了。

当然，这是满状态，也就是直线在最下方的时候，然后我们将点按照纵坐标排序，逐渐向上移动，每一次将同一行的点从树状数组删掉，然后将l,r合并。同时维护答案，跟上面的一样，用已经合并了的l,r更新一下就好了。

最后还有一种情况就是枚举一下哪一种颜色不选，然后对ans取个max，注意要在删点之前，也是用树状数组维护。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
int n,k,ans;
struct point
{
    int x,y,id,col;
}po[N];
int a[N],pos[N],val[N],last[N],Left[N],Right[N];
//int f[2][N][N];
inline int lowbit(int x)
{
    return x&(-x);
}
inline void add(int x,int v)
{
    while (x<=n+1)
    {
        val[x]+=v;
        x+=lowbit(x);
    }
}
inline int ask(int x)
{
    int ret=0;
    while (x>0)
    {
        ret+=val[x];
        x-=lowbit(x);
    }
    return ret;
}
bool cmp1(point a,point b)
{
    return a.x<b.x;
}
bool cmp2(point a,point b)
{
    return a.y<b.y;
}
inline void solve()
{
    memset(val,0,sizeof(val));
    memset(last,0,sizeof(last));
    sort(po+1,po+1+n,cmp1);
    pos[n+1]=n+1;
    fo(i,1,n)add(po[i].x,1);
    fo(i,1,n)
    {
        int l=last[po[i].col];
        Left[po[i].id]=l,Right[po[i].id]=n+1;
        last[po[i].col]=po[i].id;
        if (l)Right[l]=po[i].id;
        if (pos[l]+1<=pos[po[i].id]-1)ans=max(ans,ask(pos[po[i].id]-1)-ask(pos[l]));
    }
    fo(i,1,k)if (pos[last[i]]+1<=n)ans=max(ans,ask(n+1)-ask(pos[last[i]]));
    sort(po+1,po+1+n,cmp2);
    int j=1;
    fo(i,1,n)
    {
        while (j<=n&&po[i].y==po[j].y)add(po[j].x,-1),j++;
        Left[Right[po[i].id]]=Left[po[i].id];
        Right[Left[po[i].id]]=Right[po[i].id];
        if (pos[Right[po[i].id]]-1>=pos[Left[po[i].id]]+1)
            ans=max(ans,ask(pos[Right[po[i].id]]-1)-ask(pos[Left[po[i].id]]));
    }

}
int main()
{
    int cas=0;
    scanf("%d",&cas);
    while (cas--)
    {
        scanf("%d%d",&n,&k);
        fo(i,1,n)
        {
            scanf("%d%d%d",&po[i].x,&po[i].y,&po[i].col);
            a[i]=po[i].x;
            po[i].id=i;
            //fo(j,1,n)
            //f[0][i][j]=abs(po[i].x-po[j].x),f[1][i][j]=abs(po[i].y-po[j].y);
        }
        sort(a+1,a+1+n);
        int tot=0;
        ans=0;
        fo(i,1,n)
        po[i].x=pos[i]=lower_bound(a+1,a+1+n,po[i].x)-a;
        solve();
        fo(i,1,n)po[i].y=-po[i].y;
        solve();
        printf("%d\n",ans);
    }
}