「POJ 2182」 Lost Cows

题目链接

戳这

题目大意

\(N（2 <= N <= 8,000）\)头奶牛有\(1..N\)范围内的独特品牌.对于每头排队的牛,知道排在那头牛之前的并比那头牛的品牌小的奶牛数目.根据这些数据,输出的确切排序。

题解

令输入数组为b[i];

可以发现最后一个奶牛可以确定为b[n]+1,由最后一个又可以推出倒数第二个，由此递推就好了,每次的答案就是序列中的第b[i]+1小的数,询问完后就将这个数从序列中删除。

至于怎么选那个数，就看你用什么数据结构维护了，我用的平衡树。

code

#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
int read(){
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9')
		f=(c=='-')?-1:1,c=getchar();
	while(c>='0'&&c<='9')
		x=x*10+c-'0',c=getchar();
	return x*f;
}
int cnt,root;
struct node{
	int fa,ch[2],v,size,son;
}a[500001];
void pushup(int k){
	a[k].son=a[a[k].ch[0]].son+a[a[k].ch[1]].son+a[k].size;
}
void turn(int x){
	int y=a[x].fa,z=a[y].fa;
	int k=x==a[y].ch[1],k2=y==a[z].ch[1];
	a[z].ch[k2]=x,a[x].fa=z;
	a[y].ch[k]=a[x].ch[k^1],a[a[x].ch[k^1]].fa=y;
	a[x].ch[k^1]=y,a[y].fa=x;
	pushup(y),pushup(x);
}
void splay(int x,int u){
	while(a[x].fa!=u){
		int y=a[x].fa,z=a[y].fa;
		if(z!=u)
			(a[y].ch[0]==x)^(a[z].ch[0]==y)?turn(x):turn(y);
		turn(x);
	}
	if(u==0)
		root=x;
}
void add(int x){
	int u=root,f=0;
	while(u&&a[u].v!=x)
		f=u,u=a[u].ch[x>a[u].v];
	if(u)
		a[u].size++;
	else{
		++cnt;
		u=cnt;
		if(f)
			(x>a[f].v)?a[f].ch[1]=u:a[f].ch[0]=u;
		a[cnt].fa=f;
		a[cnt].son=1;
		a[cnt].size=1;
		a[cnt].v=x;
	}
	splay(u,0);
}
void find(int x){
	int u=root;
	if(!u)
		return ;
	while(x!=a[u].v&&a[u].ch[x>a[u].v])
		u=a[u].ch[x>a[u].v];
   splay(u,0);
}
int hou(int x){
	find(x);
	int u=root;
	if(a[u].v>x)
		return u;
	u=a[u].ch[1];
	while(a[u].ch[0])
		u=a[u].ch[0];
	return u;
}
int qian(int x){
	find(x);
	int u=root;
	if(a[u].v<x)
		return u;
	u=a[u].ch[0];
	while(a[u].ch[1])
		u=a[u].ch[1];
	return u;
}
void delet(int x){
	int l=qian(x),r=hou(x);
	splay(l,0),splay(r,l);
	int u=a[r].ch[0];
	if(a[u].size>1)
		a[u].size--,splay(u,0);
	else
		a[r].ch[0]=0;
}
int k_th(int x){
	int u=root;
	if(a[u].son<x)
		return 0;
	while(1){
		int s=a[u].ch[0];
		if(x>a[s].son+a[u].size)
			x-=a[s].son+a[u].size,u=a[u].ch[1];
		else
			if(a[s].son>=x)
				u=s;
			else
				return a[u].v;
	}
}
int hh[100001],b[10001];
int main(){
	add(-21474847);
	add(21474837);
	int n=read(),x;
	for(int i=1;i<=n;i++)
		add(i);
	for(int i=2;i<=n;i++)
		b[i]=read();
	for(int i=n;i>=1;i--)
		x=k_th(b[i]+2),hh[i]=x,delet(x);
	for(int i=1;i<=n;i++)
		printf("%d\n",hh[i]);
	return 0;
}