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

点击打开链接

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

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

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=200010;
sets[maxn];
set:: 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


你可能感兴趣的:(线段树&树状数组,线段树)