fhq treap 8pts求助(玄关)
查看原帖
fhq treap 8pts求助(玄关)
1287433
__ycy1124__楼主2024/7/20 09:49
#include<bits/stdc++.h>
using namespace std;
struct Node{
	int l,r,key,val,len;
}a[200001];
int tot=0,rot;	
void push_up(int p){
	a[p].len=a[a[p].l].len+a[a[p].r].len+1;
}
//-----------------------------------------合并 
int merge(int rot1,int rot2){
	if(rot1==0){
		return rot2;
	}
	else if(rot2==0){
		return rot1;
	}
	if(a[rot1].key<a[rot2].key){
		a[rot1].r=merge(a[rot1].r,rot2);
		push_up(rot1);
		return rot1;
	}
	else{
		a[rot2].l=merge(rot1,a[rot2].l);
		push_up(rot2);
		return rot2;
	}
}
//-----------------------------------------按权值拆 
void val_split(int now,int k,int &rot1,int &rot2){
	if(now==0){
		rot1=0;
		rot2=0;
		return;
	}
	else if(a[now].val<=k){
		rot1=now;
		val_split(a[rot1].r,k,a[rot1].r,rot2);
	}
	else{
		rot2=now;
		val_split(a[rot2].l,k,rot1,a[rot2].l);
	}
	push_up(now);
}
//-----------------------------------------按排名拆 
void k_split(int now,int k,int &rot1,int &rot2){
	if(now==0){
		rot1=0;
		rot2=0;
		return;
	}
	push_up(now);
	if(a[a[now].l].len<k){
		rot1=now;
		k_split(a[rot1].r,k-a[a[rot1].l].len-1,a[rot1].r,rot2);
	}
	else{
		rot2=now;
		k_split(a[rot2].l,k,rot1,a[rot2].l);
	}
	push_up(now);	
}
//-------------------------------------新建节点 
void New_Node(int val){
	a[++tot].key=rand()%114514+1;
	a[tot].l=0;
	a[tot].r=0;
	int rot1,rot2,now=tot;
	a[tot].len=1;
	a[tot].val=val;
	val_split(rot,val,rot1,rot2);
	rot=merge(merge(rot1,now),rot2);
}
//-------------------------------------删除 
void Erase(int val){
	int rot1,rot2,rot3,rot4;
	val_split(rot,val-1,rot1,rot2);
	k_split(rot2,1,rot3,rot4);
	rot=merge(rot1,rot4);
}
//-------------------------------------查找 
int Find_k(int val){
	int rot1,rot2;
	val_split(rot,val-1,rot1,rot2);	
	int x=a[rot1].len+1;
	rot=merge(rot1,rot2);
	return x;
}
int Find_val(int k){
	int rot1,rot2,rot3,rot4;
	k_split(rot,k-1,rot1,rot2);
	k_split(rot2,1,rot3,rot4);
	int x=a[rot3].val;
	rot=merge(rot1,merge(rot3,rot4));
	return x;
}
int Find_last(int val){
	int rot1,rot2,rot3,rot4;
	val_split(rot,val-1,rot1,rot2);
	k_split(rot1,a[rot1].len-1,rot3,rot4);
	int x=a[rot4].val;
	rot=merge(merge(rot3,rot4),rot2);
	return x;
}
int Find_next(int val){
	int rot1,rot2,rot3,rot4;
	val_split(rot,val,rot1,rot2);
	k_split(rot2,1,rot3,rot4);
	int x=a[rot3].val;
	rot=merge(rot1,merge(rot3,rot4));
	return x;
}
//------------------------------------主函数 
int w[200001];
int main(){
	srand(time(NULL));
	int n,m;
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++){
		scanf("%d",&w[i]);
	}
	int tmp=0;
	for(int i=1;i<=n;i++){
		int x;
		scanf("%d",&x);
		while(tmp<=x){
			New_Node(w[++tmp]);
		}
		printf("%d\n",Find_val(i));
	}
	return 0;
}
2024/7/20 09:49
加载中...