Lost Cows（思维 + 线段树/树状数组 + 二分）

题目链接
acwing原题：AcWing 244. 谜一样的牛

分析

模拟发现从后往前模拟是可行的；
每次相当于是从1 2 3 … n中找到第a[i]+1小的且没被用过的数；

做法1：树外二分 + 树状数组
用树状数组维护前缀和，初始值每个点为1；
每次二分当前位置，query看是否=a[i]+1；
找到的话该位置-1就行；
时间复杂度O(nlognlogn)

做法2：线段树 + 树上二分
转换一下问题：
由于是给一个有序序列1 2 3 … n，所以第k小数其实就是第k个数；
问题就转换成，每次找第k个数，然后删掉；
这不就是前几天写的问题嘛！
线段树维护一个有效值 和 一个权值，然后树上二分就行
时间复杂度O(n*logn)

代码

int a[maxn];
int n;

struct node
{	
	int l,r;
	int v;
	int flag;	//是否有效

}tr[maxn*4];

int ans[maxn];
int b[maxn];

void pushup(int u)
{
	tr[u].flag = tr[u<<1].flag + tr[u<<1|1].flag;
}

void build(int u,int l,int r)
{
	if(l == r)
	{
		tr[u] = {l,r,b[l],1};
		return ;
	}
	
	tr[u] = {l,r};
	
	int mid = l + r >> 1;
	
	build(u<<1,l,mid);
	build(u<<1|1,mid+1,r);
	
	pushup(u);
}

void del(int u,int idx)	//删除
{
	if(tr[u].r == idx && tr[u].l == idx)
	{
		tr[u].flag = 0;
		return ;
	}
	
	int mid = tr[u].l + tr[u].r >> 1;
	
	if(idx <= mid) del(u<<1,idx);
	else del(u<<1|1,idx);
	
	pushup(u);
}

node query(int u,int k)
{
	if(tr[u].l == tr[u].r) return tr[u];
	
	if(tr[u<<1].flag >= k) return query(u<<1,k);
	else return query(u<<1|1,k-tr[u<<1].flag);
}

int main()
{
	ire(n);
	for(int i=2;i<=n;i++) ire(a[i]);
	
	for(int i=1;i<=n;i++) b[i] = i;
	
	build(1,1,n);
	
	for(int i=n;i>=1;i--)
	{
		int k = a[i] + 1;
		
		ans[i] = query(1,k).v;
		
		del(1,ans[i]);
	}
	
	for(int i=1;i<=n;i++) cout<<ans[i]<<endl;
	
	return 0;
}