bzoj4548: 小奇的糖果【链表+扫描线+树状数组】

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

解题思路：

注意是线段，不是直线，否则就没意思了。
最优解肯定是最相邻几个相同颜色的点围成的矩形，一共有三种矩形：
1.上下不封顶（线段极高或极低），左右边界是两个相邻的同色点，这个扫一遍用前缀和相减即可。
2.下界一点封底，左右边界是和下界点同色且y大于它的相邻点，这种情况用双向链表记录每个点的前驱后继同色点，按y从小到大扫描，前缀和用树状数组维护，每扫描一个点就把其从链表和前缀和中删去。
3、上界、左右界封住，方法同2，把y坐标取反即可。

#include<bits/stdc++.h>
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=100005;
struct point{int x,y,c,id;}p[N];
int T,n,k,ans;
int l[N],r[N],bit[N],b[N],pre[N],w[N];

inline bool cmpx(const point &a,const point &b){return a.x<b.x;}
inline bool cmpy(const point &a,const point &b){return a.y<b.y;}

inline void add(int i,int val)
{
    for(;i<=n+1;i+=i&(-i))bit[i]+=val;
}

inline int query(int i)
{
    int res=0;
    for(;i;i-=i&(-i))res+=bit[i];
    return res;
}

inline void update(int l,int r)
{
    ans=max(ans,query(r)-query(l-1));
}

void solve()
{
    memset(bit,0,sizeof(bit));
    memset(pre,0,sizeof(pre));
    w[0]=0,w[n+1]=n+1;
    sort(p+1,p+n+1,cmpx);
    for(int i=1;i<=n;i++)add(p[i].x,1);
    int x,y;
    for(int i=1;i<=n;i++)
    {
        x=p[i].id,y=pre[p[i].c];
        l[x]=y,r[x]=n+1;
        if(y)r[y]=x;
        update(w[y]+1,w[x]-1);
        pre[p[i].c]=x;
    }
    for(int i=1;i<=k;i++)update(w[pre[i]]+1,n+1);
    sort(p+1,p+n+1,cmpy);
    for(int i=1,j=1;i<=n;i++)
    {
        x=p[i].id;
        while(j<=n&&p[j].y==p[i].y)add(p[j++].x,-1);
        update(w[l[x]]+1,w[r[x]]-1);
        l[r[x]]=l[x],r[l[x]]=r[x];
    }
}

int main()
{
    //freopen("lx.in","r",stdin);
    T=getint();
    while(T--)
    {
        ans=0;n=getint(),k=getint();
        for(int i=1;i<=n;i++)
            p[i].x=b[i]=getint(),p[i].y=getint(),p[i].c=getint(),p[i].id=i;
        sort(b+1,b+n+1);
        for(int i=1;i<=n;i++)p[i].x=lower_bound(b+1,b+n+1,p[i].x)-b,w[i]=p[i].x;
        solve();
        for(int i=1;i<=n;i++)p[i].y=-p[i].y;
        solve();
        printf("%d\n",ans);
    }
    return 0;
}