CodeForces - 12D Ball【离散化】【线段树】

题目链接：https://codeforces.com/contest/12/problem/D

先按b降序排列，对i离散化，查询(i,+∞)中最大的r是否比当前r大

#include <iostream>
#include <vector>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
static const int MAXN=5e5+10;
struct Node{
    int b,i,r;
    bool operator<(const Node &o)const{
        return b>o.b;
    }
}lady[MAXN];
vector<int> alls;
int n,maxi;
int find(int x)
{
    return lower_bound(alls.begin(),alls.end(),x)-alls.begin()+1;
}
struct SegTree{
    int l,r;
    int dat;
}tr[MAXN<<2];
void push_up(int u) { tr[u].dat=max(tr[u<<1].dat,tr[u<<1|1].dat); }
void build(int u,int l,int r)
{
    tr[u].l=l; tr[u].r=r;
    if(l==r) { tr[u].dat=-1; return; }
    int mid=l+r>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
    push_up(u);
}
void upd(int u,int x,int v)
{
    if(tr[u].l==tr[u].r)
    {
        tr[u].dat=max(v,tr[u].dat);
        return;
    }
    int mid=tr[u].l+tr[u].r>>1;
    if(x<=mid) upd(u<<1,x,v);
    else upd(u<<1|1,x,v);
    push_up(u);
}
int query(int u,int l,int r)
{
    if(l<=tr[u].l && tr[u].r<=r) return tr[u].dat;
    int mid=tr[u].l+tr[u].r>>1;
    int v=-(1<<30);
    if(l<=mid) v=max(v,query(u<<1,l,r));
    if(r>mid) v=max(v,query(u<<1|1,l,r));
    return v;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&lady[i].b);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&lady[i].i);
        alls.push_back(lady[i].i);
    }
    for(int i=0;i<n;i++) scanf("%d",&lady[i].r);
    sort(lady,lady+n);
    alls.push_back(0);
    sort(alls.begin(),alls.end());
    alls.erase(unique(alls.begin(),alls.end()),alls.end());
    build(1,1,alls.size()+1);
    int res=0;
    queue<int> wait;
    for(int i=0;i<n;i++)
    {
        while(!wait.empty() && lady[wait.front()].b>lady[i].b)
		{
            upd(1,find(lady[wait.front()].i),lady[wait.front()].r);
            wait.pop();
        }
        if(query(1,find(lady[i].i)+1,alls.size()+1)>lady[i].r) res++;
        wait.push(i);
    }
    printf("%d\n",res);
    return 0;
}