coderforces round 19D线段树+离散化处理

来源:互联网 发布:通信行业 知乎 编辑:程序博客网 时间:2024/06/05 16:39

点击打开链接

题意:共有三种操作,add:向平面中加入一个点x,y,remove:将平面中的一个点删除,find x,y查询平面内严格大于x,y的点,要求这个点横坐标越小越好,然后再保证纵坐标越小越好,不存在这样的点输出-1。

思路:因为x,y的范围很大,所以先将x坐标的值离散化,然后以x的位置和对应y的值保存到set中,线段树节点保存区间最大值,查询时也查询大于x的位置,然后二分求得y的位置即可。下面有注释。

#include <set>#include <stdio.h>#include <string.h>#include <stdlib.h>#include <iostream>#include <algorithm>using namespace std;typedef long long ll;const int inf=0x3f3f3f3f;const int maxn=200010;set<int>s[maxn];set<int>:: iterator ite;int num[maxn*4],x[maxn],y[maxn],t[maxn];char str[maxn][20];void pushup(int node){    num[node]=max(num[node<<1],num[node<<1|1]);    //节点保存y轴的最大值}void update(int pos,int yy,int op,int le,int ri,int node){    //pos为离散后x的值,yy为x对应的y轴坐标    if(le==ri){        //op只为1添加,值为0删除        if(op){            num[node]=max(num[node],yy);            s[le].insert(yy);        }else{            s[le].erase(yy);            num[node]=*(s[le].end());//因为可能一个x对应了多个y,保证最大值,所以将num[node]的值改为x现在对应的最大值        }        return ;    }    int t=(le+ri)>>1;    if(pos<=t) update(pos,yy,op,le,t,node<<1);    else update(pos,yy,op,t+1,ri,node<<1|1);    pushup(node);}int query(int l,int r,int yy,int le,int ri,int node){    if(num[node]<=yy) return -1;    if(l<=le&&ri<=r){        if(le==ri) return le;        int t=(le+ri)>>1;        if(num[node<<1]>yy) return query(l,r,yy,le,t,node<<1);        //这应该不用讲了,找最小的点大于查询的点,所以尽可能的先满足左侧的点        //只要num[node<<1]大于yy,说明左儿子区间内有大于yy的点,并且满足尽量小的条件        if(num[node<<1|1]>yy) return query(l,r,yy,t+1,ri,node<<1|1);        return -1;    }    int t=(le+ri)>>1;    if(l<=t){        int k=query(l,r,yy,le,t,node<<1);        if(k!=-1) return k;    }    if(r>t){        int k=query(l,r,yy,t+1,ri,node<<1|1);        if(k!=-1) return k;    }    return -1;}int main(){    int n;    while(scanf("%d",&n)!=-1){        for(int i=0;i<maxn;i++) s[i].clear();        memset(num,0,sizeof(num));        int k=0;        for(int i=0;i<n;i++){            scanf("%s%d%d",str[i],&x[i],&y[i]);            t[k++]=x[i];        }        sort(t,t+k);        int len=unique(t,t+k)-t;        //将x坐标进行离散化,毕竟10的9次方        for(int i=0;i<n;i++){            int pos=lower_bound(t,t+len,x[i])-t+1;            //当前x的位置            if(str[i][0]=='a') update(pos,y[i],1,1,len,1);            else if(str[i][0]=='r') update(pos,y[i],0,1,len,1);            else{                int ans=query(pos+1,len,y[i],1,len,1);                //pos+1因为题目要求严格大于询问的坐标,所以从下一个开始查询                if(ans==-1) printf("-1\n");                else printf("%d %d\n",t[ans-1],*(s[ans].upper_bound(y[i])));                //找到了x的坐标,但因为t数组从0开始,结果减去1,y坐标用二分进行快速查询            }        }    }    return 0;}


0 0
原创粉丝点击