求助(玄关)有1点点注释
查看原帖
求助(玄关)有1点点注释
1199534
ycy1124楼主2024/7/2 14:23
#include<bits/stdc++.h>
using namespace std;
struct Node{
	int l,r,key,val,len;
}a[100001];
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(a[rot2].l,rot1);
		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[rot1].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){
	srand(time(NULL));
	a[++tot].key=rand()%114514+1;
	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(merge(rot1,rot2),rot4);
}
//-------------------------------------查找 
void Find_k(int val){
	int rot1,rot2;
	val_split(rot,val-1,rot1,rot2);	
	printf("%d\n",a[rot1].len+1);
	rot=merge(rot1,rot2);
}
void Find_val(int k){
	int rot1,rot2,rot3,rot4;
	k_split(rot,k-1,rot1,rot2);
	k_split(rot2,1,rot3,rot4);
	printf("%d\n",a[rot3].val);
	rot=merge(rot1,merge(rot3,rot4));
}
void 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);
	printf("%d\n",a[rot4].val);
	rot=merge(merge(rot3,rot4),rot2);
}
void Find_next(int val){
	int rot1,rot2,rot3,rot4;
	val_split(rot,val-1,rot1,rot2);
	k_split(rot2,1,rot3,rot4);
	printf("%d\n",a[rot3].val);
	rot=merge(rot1,merge(rot3,rot4));
}
//------------------------------------主函数 
int main(){
	srand(time(NULL));
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		int op;
		scanf("%d",&op);
		int val;
		scanf("%d",&val);
		if(op==1){
			New_Node(val);
		}
		else if(op==2){	
			Erase(val);
			tot--;
		}
		else if(op==3){
			Find_k(val);
		}
		else if(op==4){
			Find_val(val);
		}
		else if(op==5){
			Find_last(val);
		}
		else{
			Find_next(val);
		}
	}
	return 0;
}
2024/7/2 14:23
加载中...