poj 2761 treap
利用区间的有序性才可以用treap 其实这题 线段树和树状数组时更好的选择 稍后补上
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
const int inf=~0U>>1;
class treap
{
struct node
{
int val,key,size;
node *c[2];
node(int v,node *x){val=v,c[0]=c[1]=x,size=1;key=rand()-1;}
void rz(){size=c[0]->size+c[1]->size+1;}
}*root,*null;
void ro(node *&x,bool d)
{
node *t=x->c[d];
x->c[d]=t->c[!d];
t->c[!d]=x;
x->rz();t->rz();x=t;
}
void insert(node *&x,int v)
{
if(x==null) {x=new node(v,null);return;}
// if(x->val==v) return ;
bool d=(v>=x->val);
insert(x->c[d],v);
if(x->c[d]->key<x->key)
ro(x,d); else x->rz();
}
void dele(node*&x,int v)
{
if(x==null) return ;
if(x->val==v)
{
bool d=x->c[0]->key>x->c[1]->key;
if(x->c[d]==null)
{
x=null;
return ;
}
ro(x,d);
dele(x->c[!d],v);
}
else
{
bool d=v>x->val;
dele(x->c[d],v);
}
x->rz();
}
int select(node *x,int k)
{
if(x->c[0]->size==k) return x->val;
else if(x->c[0]->size>k) return select(x->c[0],k);
else return select(x->c[1],k-x->c[0]->size-1);
}
int ran(node *x,int v)
{
if(x==null) return 0;
if(x->val==v) return x->c[0]->size;
if(x->val>v) return ran(x->c[0],v);
else return ran(x->c[1],v)+1+x->c[0]->size;
}
void pr(node *x) {if(x!=null) {pr(x->c[0]);cout<<x->val<<" ";pr(x->c[1]);}}
public:
treap(){null=new node(0,0);null->size=0;null->key=inf;root=null;}
void ins(int v){insert(root,v);}
void del(int v){dele(root,v);}
int rank(int v){return ran(root,v);}
int find_k(int k){if(root->size<k) return -inf;return select(root,k-1);}
void pr(){pr(root);puts("");}
};
int n,m;
int val[111111];
struct query
{
int l,r,id,k;
bool operator <(query q)
{
return l<q.l||(l==q.l&&r<q.r);
}
}a[55555];
int ans[55555];
template <class T>
void quicksort(T *p,int s,int e)
{
if(s<e)
{
int i=s,j=e;T tmp=p[e];
while(i<j)
{
while(i<j&&p[i]<tmp) ++i;
if(i<j) p[j--]=p[i];
while(i<j&&tmp<p[j]) --j;
if(i<j) p[i++]=p[j];
}
p[i]=tmp;
quicksort(p,s,i-1);
quicksort(p,i+1,e);
}
}
int main()
{
while(scanf("%d%d",&n,&m)==2)
{
treap T;
for(int i=1;i<=n;++i)
{
scanf("%d",val+i);
}
for(int i=1;i<=m;++i)
{
scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].k);
a[i].id=i;
}
quicksort(a,1,m);
for(int i=1;i<=m;++i)
{
//cout<<a[i].l<<" "<<a[i].r<<" "<<a[i].k<<" "<<a[i].id;
int d=(a[i-1].r<a[i].l-1)?a[i-1].r:a[i].l-1;
int in=(a[i-1].r+1>a[i].l)?a[i-1].r+1:a[i].l;
// cout<<" "<<d<<" "<<in<<endl;
if(i>1) for(int j=a[i-1].l;j<=d;++j)
T.del(val[j]);
for(int j=in;j<=a[i].r;++j)
T.ins(val[j]);
ans[a[i].id]=T.find_k(a[i].k);
// T.pr();
}
for(int i=1;i<=m;++i)
printf("%d\n",ans[i]);
}
return 0;
}
我次奥 换成系统的sort 直接变成 3000ms
线段树版本
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define N 111111
using namespace std;
int sum[N<<2];
void pup(int rt)
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int p,int add,int l,int r,int rt)
{
if(l==r)
{
sum[rt]+=add;
return;
}
int mid=(l+r)>>1;
if(p<=mid) update(p,add,lson);
else update(p,add,rson);
pup(rt);
}
int que(int k,int l,int r,int rt)
{
if(l==r) return l;
int mid=(l+r)>>1;
if(k<=sum[rt<<1])
return que(k,lson);
else
return que(k-sum[rt<<1],rson);
}
struct ps
{
int id,val;
}a[111111];
int va[111111];
int cmp(ps p,ps q){return p.val<q.val;}
int n,m;
struct pp
{
int l,r,id,k;
}q[55555];
int cmp2(pp p,pp q){return p.l<q.l;}
int ans[55555];
int main()
{
while(scanf("%d%d",&n,&m)==2)
{
for(int i=1;i<=n;++i)
{
scanf("%d",&a[i].val);
a[i].id=i;
}
sort(a+1,a+1+n,cmp);
for(int i=1;i<=n;++i)
va[a[i].id]=i;
memset(sum,0,sizeof(sum));
for(int i=1;i<=m;++i)
{
scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
q[i].id=i;
}
sort(q+1,q+1+m,cmp2);
for(int i=q[1].l;i<=q[1].r;++i)
update(va[i],1,1,n,1);
ans[q[1].id]=que(q[1].k,1,n,1);
for(int i=2;i<=m;++i)
{
int d,in;
d=(q[i-1].r<q[i].l-1)?q[i-1].r:q[i].l-1;
in=(q[i-1].r+1<q[i].l)?q[i].l:q[i-1].r+1;
for(int j=q[i-1].l;j<=d;++j)
update(va[j],-1,1,n,1);
for(int j=in;j<=q[i].r;++j)
update(va[j],1,1,n,1);
ans[q[i].id]=que(q[i].k,1,n,1);
}
for(int i=1;i<=m;++i)
printf("%d\n",a[ans[i]].val);
}
return 0;
}
下面是树状数组
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 111111
using namespace std;
int sum[N];
struct ps
{
int id,val;
}a[111111];
int va[111111];
int cmp(ps p,ps q){return p.val<q.val;}
int n,m;
struct pp
{
int l,r,id,k;
}q[55555];
int cmp2(pp p,pp q){return p.l<q.l;}
int ans[55555];
int lowbit(int x){return x&(x^(x-1));}
void update(int p,int add)
{
for(;p<=n;p+=lowbit(p))
sum[p]+=add;
}
int qu(int p)
{
int res;
for(res=0;p;p-=lowbit(p))
res+=sum[p];
return res;
}
int que(int k)
{
int l=1,r=n,mid;
for(;l<=r;)
{
mid=(l+r)>>1;
int res=qu(mid);
if(res>=k) r=mid-1;
else l=mid+1;
}
return l;
}
int main()
{
while(scanf("%d%d",&n,&m)==2)
{
for(int i=1;i<=n;++i)
{
scanf("%d",&a[i].val);
a[i].id=i;
}
sort(a+1,a+1+n,cmp);
for(int i=1;i<=n;++i)
va[a[i].id]=i;
memset(sum,0,sizeof(sum));
for(int i=1;i<=m;++i)
{
scanf("%d%d%d",&q[i].l,&q[i].r,&q[i].k);
q[i].id=i;
}
sort(q+1,q+1+m,cmp2);
for(int i=q[1].l;i<=q[1].r;++i)
update(va[i],1);
ans[q[1].id]=que(q[1].k);
for(int i=2;i<=m;++i)
{
int d,in;
d=(q[i-1].r<q[i].l-1)?q[i-1].r:q[i].l-1;
in=(q[i-1].r+1<q[i].l)?q[i].l:q[i-1].r+1;
for(int j=q[i-1].l;j<=d;++j)
update(va[j],-1);
for(int j=in;j<=q[i].r;++j)
update(va[j],1);
ans[q[i].id]=que(q[i].k);
}
for(int i=1;i<=m;++i)
printf("%d\n",a[ans[i]].val);
}
return 0;
}