转载于:百度百科
支持以下操作
支持以下操作
1 x 若x不存在,插入x
2 x 若x存在,删除x
3 输出当前最小值,若不存在输出-1
4 输出当前最大值,若不存在输出-1
5 x 输出x的前驱,若不存在输出-1
6 x 输出x的后继,若不存在输出-1
7 x 若x存在,输出1,否则输出-1
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<map>
#include<set>
#include<vector>
#include<queue>
#define inf 1000000000
using namespace std;
int n,m;
struct seg{int l,r,v;}t[3000005];
void build(int k,int l,int r)
{
t[k].l=l;t[k].r=r;
if(l==r)return;
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
}
int mn(int k)
{
if(!t[k].v)return -1;
int l=t[k].l,r=t[k].r;
if(l==r)return l;
if(t[k<<1].v)return mn(k<<1);
else return mn(k<<1|1);
}
int mx(int k)
{
if(!t[k].v)return -1;
int l=t[k].l,r=t[k].r;
if(l==r)return l;
if(t[k<<1|1].v)return mx(k<<1|1);
else return mx(k<<1);
}
void insert(int k,int val)
{
int l=t[k].l,r=t[k].r;
if(l==r){t[k].v=1;return;}
int mid=(l+r)>>1;
if(val<=mid)insert(k<<1,val);
else insert(k<<1|1,val);
t[k].v=t[k<<1].v+t[k<<1|1].v;
}
int find(int k,int val)
{
int l=t[k].l,r=t[k].r;
if(l==r)
{
if(t[k].v)return 1;
return -1;
}
int mid=(l+r)>>1;
if(val<=mid)return find(k<<1,val);
else return find(k<<1|1,val);
}
void del(int k,int val)
{
int l=t[k].l,r=t[k].r;
if(l==r){t[k].v=0;return;}
int mid=(l+r)>>1;
if(val<=mid)del(k<<1,val);
else del(k<<1|1,val);
t[k].v=t[k<<1].v+t[k<<1|1].v;
}
int findpr(int k,int val)
{
if(val<0)return -1;
if(!t[k].v)return -1;
int l=t[k].l,r=t[k].r;
if(l==r)return l;
int mid=(l+r)>>1;
if(val<=mid)return findpr(k<<1,val);
else
{
int t=findpr(k<<1|1,val);
if(t==-1)return mx(k<<1);
else return t;
}
}
int findsu(int k,int val)
{
if(!t[k].v)return -1;
int l=t[k].l,r=t[k].r;
if(l==r)return l;
int mid=(l+r)>>1;
if(val>mid)return findsu(k<<1|1,val);
else
{
int t=findsu(k<<1,val);
if(t==-1)return mn(k<<1|1);
else return t;
}
}
int main()
{
scanf("%d %d",&n,&m);
build(1,0,n);
int opt,x;
for(int i=1;i<=m;i++)
{
scanf("%d",&opt);
switch(opt)
{
case 1:scanf("%d",&x);if(find(1,x)==-1)insert(1,x);break;
case 2:scanf("%d",&x);if(find(1,x)==1)del(1,x);break;
case 3:printf("%d\n",mn(1));break;
case 4:printf("%d\n",mx(1));break;
case 5:scanf("%d",&x);printf("%d\n",findpr(1,x-1));break;
case 6:scanf("%d",&x);printf("%d\n",findsu(1,x+1));break;
case 7:scanf("%d",&x);printf("%d\n",find(1,x));break;
}
}
return 0;
}