2020牛客寒假算法基础集训营4 I 匹配星星

https://ac.nowcoder.com/acm/contest/3005/I

又做麻烦了，悲催。。。

将所有星星按x坐标为第一关键字，z为第二关键字排好序

那么一个z=1的星星匹配的是x比它小的，y比他小但又尽可能大的星星

用线段树维护所有的y坐标

如果z=0，直接把y坐标插入线段树

如果z=1，在线段树中找小于它的y坐标，y又最大的那个星星，线段树中删走

题解直接用的set，而且在x相同时先处理z=1再处理z=0更方便。唉。。。

#include<cstdio>
#include<algorithm>

using namespace std;

#define N 100002

struct node
{
    int x,y;
}e0[N],e1[N];

int m,has[N];
int sum[N<<],tot;

bool cmp(node p,node q)
{
    return p.x<q.x;
}

void change(int k,int l,int r,int pos,int x)
{
    sum[k]+=x;
    if(l==r) return;
    int mid=l+r>>;
    if(pos<=mid) change(k<<,l,mid,pos,x);
    else change(k<<|,mid+,r,pos,x);
}

void query2(int k,int l,int r,int opl,int opr)
{
    if(l>=opl && r<=opr)
    {
        tot+=sum[k];
        return;
    }
    int mid=l+r>>;
    if(opl<=mid) query2(k<<,l,mid,opl,opr);
    if(opr>mid) query2(k<<|,mid+,r,opl,opr);
}

int query(int k,int l,int r,int pos)
{
    if(l==r)
        if(sum[k]) return l;
        else return ;
    int mid=l+r>>;
    tot=;
    query2(,,m,mid+,pos);
    if(tot) return query(k<<|,mid+,r,pos);
    return query(k<<,l,mid,pos);
}

int main()
{
    int n,x,y,z,n0=,n1=;
    scanf("%d",&n);
    for(int i=;i<=n;++i)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(z) e1[++n1].x=x,e1[n1].y=y;
        else e0[++n0].x=x,e0[n0].y=y;
        has[i]=y;
    }
    sort(has+,has+n+);
    m=unique(has+,has+n+)-has-;
    sort(e0+,e0+n0+,cmp);
    sort(e1+,e1+n1+,cmp);
    e0[n0+].x=1e9+;
    int now0=,now1=,i,mx,ans=;
    while(now1<=n1 && e1[now1].x<=e0[].x) now1++;
    while(now0<=n0)
    {
        for(i=now0;i<=n0 && e0[i].x==e0[now0].x;++i) change(,,m,lower_bound(has+,has+m+,e0[i].y)-has,);
        while(now1<=n1 && e1[now1].x<=e0[now0+].x)
        {
            mx=query(,,m,lower_bound(has+,has+m+,e1[now1].y)-has-);
            if(mx) change(,,m,mx,-),ans++;
            now1++;
        }
        now0=i;
    }
    printf("%d",ans);
    return ;
}