CodeForces - 19D Deletion of Repeats【线段树】【离散化】

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

线段树套set,2800有点高了

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
static const int MAXN=2e5+10;
struct Cmd{
    int type,x,y;
}cmd[MAXN];
vector<int> Px;
set<int> Py[MAXN];
int n;
char str[MAXN];
int x,y;
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=0; 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 y,int op)
{
    if(tr[u].l==tr[u].r)
    {
        if(op==0) Py[x].insert(y);
        else Py[x].erase(y);
        if(Py[x].size()==0) tr[u].dat=0;
        else tr[u].dat=*(--Py[x].end());
        return;
    }
    int mid=tr[u].l+tr[u].r>>1;
    if(x<=mid) upd(u<<1,x,y,op);
    else upd(u<<1|1,x,y,op);
    push_up(u);
}
int query(int u,int l,int r,int x,int y)
{
    if(l==r)
    {
        if(l>x && tr[u].dat>y) return l;
        return 0;
    }
    int tmp=0;
    int mid=l+r>>1;
    if(x<mid && tr[u<<1].dat>y) tmp=query(u<<1,l,mid,x,y);
    if(tmp==0 && tr[u<<1|1].dat>y) tmp=query(u<<1|1,mid+1,r,x,y);
    return tmp;
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%s%d%d",str,&x,&y);
        if(str[0]=='a') cmd[i]={0,x,y};
        else if(str[0]=='r') cmd[i]={1,x,y};
        else cmd[i]={2,x,y};
        Px.push_back(x);
    }
    sort(Px.begin(),Px.end());
    Px.erase(unique(Px.begin(),Px.end()),Px.end());
    build(1,1,Px.size());
    for(int i=0;i<n;i++)
    {
        int xx=lower_bound(Px.begin(),Px.end(),cmd[i].x)-Px.begin()+1;
        if(cmd[i].type==0) upd(1,xx,cmd[i].y,0);
        else if(cmd[i].type==1) upd(1,xx,cmd[i].y,1);
        else if(cmd[i].type==2)
        {
            int ans=query(1,1,Px.size(),xx,cmd[i].y);
            if(!ans) puts("-1");
            else printf("%d %d\n",Px[ans-1],*Py[ans].upper_bound(cmd[i].y));
        }
    }
    return 0;
}
©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页