题意:给出n组数,第一个数是id。第二个数是级别。每输入一个。输出这个人和哪个人打架,这个人会找和他级别最相近的人打,假设有两个人级别和他相差的一样多,他就会选择级别比他小的打架。
思路:用treap完毕,能够用STL水过,但要练Treap就写了平衡树的。对于每一个人的等级,我们找到他的等级的排名t。然后找第t-1大的数和第t+1大的数。然后进行比較一下。讨论后输出,PS:没有好得模版,仅仅能在网上学了一个还能够接受的.......
#include <stdio.h> #include <string.h> #include <stdlib.h> #include <iostream> #include <algorithm> using namespace std; typedef long long ll; const int inf=0x3f3f3f3f; const int maxn=100010; class Treap{ public: struct Treap_Node{ Treap_Node *left,*right; int value,fix,weight,size;//fix为修正值。是随机产生的,保证二叉树不会退化的重要环节,wweight为权值。size为子树大小 }*root,*null; Treap(){ null=new struct Treap_Node; null->size=0; null->weight=0; null->value=inf; null->left=null; null->right=null; null->fix=inf; root=null; } void Treap_Print(Treap_Node *P){//从小到大输出 if(P!=null){ Treap_Print(P->left); printf("%d ",P->value); Treap_Print(P->right); } } void Treap_Left_Rotate(Treap_Node *&a){//左旋之后仍不改变二叉树性质 Treap_Node *b=a->right; a->right=b->left; b->left=a; b->size=a->size; a->size=a->left->size+a->right->size+a->weight; a=b; } void Treap_Right_Rotate(Treap_Node *&a){//右旋之后仍不改变二叉树性质 Treap_Node *b=a->left; a->left=b->right; b->right=a; b->size=a->size; a->size=a->left->size+a->right->size+a->weight; a=b; } int Treap_Find(Treap_Node *P,int value){//查找有没有value这个数 if(P==null) return 0; if(P->value==value) return 1; else if(value<P->value) return Treap_Find(P->left,value); else return Treap_Find(P->right,value); } void Treap_Insert(Treap_Node *&P,int value){//插入一个数 if(P==null){ P=new Treap_Node; P->left=P->right=null;//左右儿子均为空 P->value=value; P->fix=rand(); P->weight=1; P->size=1; }else if(value==P->value){ P->weight++; } else if(value<P->value){ Treap_Insert(P->left,value); if(P->left->fix<P->fix) Treap_Right_Rotate(P); }else{ Treap_Insert(P->right,value); if(P->right->fix<P->fix) Treap_Left_Rotate(P); } P->size=P->left->size+P->right->size+P->weight; } void Treap_Delete(Treap_Node *&P,int value){//删除一个数 if(P==null) return ; if(value<P->value) Treap_Delete(P->left,value); else if(value>P->value) Treap_Delete(P->right,value); else if(P->weight>1) P->weight--; else if((P->left==NULL)&&(P->right==NULL)){ delete P; P=NULL; }else{ if(P->left->fix<P->right->fix) Treap_Left_Rotate(P); else Treap_Right_Rotate(P); Treap_Delete(P,value); } P->size=P->left->size+P->right->size+P->weight; } int Treap_pred(Treap_Node *P,int value,Treap_Node *optimal){ if(P==null||value==P->value) return optimal->value; if(P->value<value) return Treap_pred(P->right,value,P); else return Treap_pred(P->left,value,optimal); } int Treap_succ(Treap_Node *P,int value,Treap_Node *optimal){ if(P==null||value==P->value) return optimal->value; if(P->value>value) return Treap_succ(P->left,value,P); else return Treap_succ(P->right,value,optimal); } int Treap_Findkth(Treap_Node *P,int k){//求第K大的数 if(P==null) return 0; int t=P->left->size; if(k<t+1) return Treap_Findkth(P->left,k); else if(k>t+P->weight) return Treap_Findkth(P->right,k-(t+P->weight)); else return P->value; } int Treap_Rank(Treap_Node *P,int value,int cur){ int t=P->left->size; if(value==P->value) return t+cur+1; else if(value<P->value) return Treap_Rank(P->left,value,cur); else return Treap_Rank(P->right,value,t+cur+P->weight); } void Treap_erase(Treap_Node *&P) { if(P->left!=null) Treap_erase(P->left); if(P->right!=null) Treap_erase(P->right); delete P; } }; int id[5000010]; int main(){ int n,a,b; while(scanf("%d",&n)!=-1){ if(n==0) break; Treap tree; int ans1,ans2; memset(id,0,sizeof(id)); scanf("%d%d",&a,&b); printf("%d 1 ",a); id[b]=a;tree.Treap_Insert(tree.root,b); for(int i=1;i<n;i++){ scanf("%d%d",&a,&b); id[b]=a; tree.Treap_Insert(tree.root,b); int ans; int k=tree.Treap_Rank(tree.root,b,0); ans1=tree.Treap_Findkth(tree.root,k-1); ans2=tree.Treap_Findkth(tree.root,k+1); if(!ans1) ans=ans2; else if(!ans2) ans=ans1; else{ if(b-ans1<=ans2-b){ ans=ans1; }else ans=ans2; } printf("%d %d ",a,id[ans]); } } return 0; }