bzoj3658 树状数组+双向链表

题目：http://www.lydsy.com/JudgeOnline/problemstatus.php?id=3658
大意：平面上有n个点，每个点有k种颜色中的一个。你可以选择一条水平的线段获得在其上方或其下方的所有点。请求出你最多能够得到多少点，使得获得的点并不包含所有的颜色。

————————————————————————————————————————————————–
用链表记录每个位置相同颜色的前一个和后一个，先考虑选择上方的（另一种将纵坐标倒过来就行了）。将纵坐标排序，每次删除一行，将相同颜色的前一个和后一个加入树状数组，在链表中删除这个点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
struct pt { int x,y,c,id; }a[100050];
int n,K,T,sum,ans;
int d[100050],X[100050],last[100050],pre[100050],nex[100050];
inline bool cmp(pt a,pt b)
{ return a.x<b.x; }
inline bool cmp1(pt a,pt b)
{ return a.y<b.y; }
inline int read()
{
    char ch;
    while (ch=getchar(),ch<'0'||ch>'9');
    int r=ch-'0';
    while (ch=getchar(),ch>='0'&&ch<='9') r=r*10-'0'+ch;
    return r;
}
inline void Add(int x,int v)
{
    for (int i=x;i<=sum;i+=i & (-i))
        d[i]+=v;
}
inline int Ask(int x)
{
    int r=0;
    for (int i=x;i;i-=i & (-i))
        r+=d[i];
    return r;
}
inline void Up(int l,int r)
{
    ans=max(ans,Ask(r)-Ask(l-1));
}
inline void Down()
{
    memset(d,0,sizeof(d));
    for (int i=1;i<=n;++i) Add(a[i].x,1);
    memset(last,0,sizeof(last));
    for (int i=1;i<=n;++i)  {
        a[i].id=i; X[i]=a[i].x;
        pre[i]=last[a[i].c];    nex[i]=n+1;
        if (last[a[i].c]) nex[last[a[i].c]]=i;
        Up(a[pre[i]].x+1,a[i].x-1);
        last[a[i].c]=i;
    }
    X[n+1]=n+1;
    for (int i=1;i<=K;++i) Up(X[last[i]]+1,n+1);
    sort(a+1,a+n+1,cmp1);
    for (int i=1,j=1;i<=n;++i) {
        int t=a[i].id;
        while (j<=n&&a[j].y==a[i].y) {
            Add(a[j].x,-1); ++j;
        }
        pre[nex[t]]=pre[t]; 
        nex[pre[t]]=nex[t];
        //cerr<<nex[t]<<endl;
        Up(X[pre[t]]+1,X[nex[t]]-1);
    }
}
int main()
{
    scanf("%d",&T);
    while (T--) {
        n=read(); K=read();
        for (int i=1;i<=n;++i) {
            a[i].x=read(); a[i].y=read(); a[i].c=read();
        }
        sort(a+1,a+n+1,cmp);

        sum=1;
        for (int i=1;i<=n;++i) {
            if (a[i].x!=a[i+1].x) a[i].x=sum++; else a[i].x=sum;
        }
        sum=n+1;
        ans=0;
        Down();
        for (int i=1;i<=n;++i) a[i].y=-a[i].y;
        sort(a+1,a+n+1,cmp);
        Down();
        printf("%d\n",ans);
    }
}