趁热来一发线段树,主要是联系一下线段树的基本操作,写了两道题之后确实感觉手感好了很多,这道题的思路是,从后向前挨个数,最后一个牛总是可以判断是第几个,然后就把相应的位置占住,以后再数的时候就不能数已经占住的位置了,说的不是很清楚,但是代码的思路比较清晰:
#include <iostream>
#include <cstdio>
#include <memory.h>
using namespace std;
#define maxn 8050
struct node{
int left;
int right;
int len;
};
node tree[maxn<<2]={0};
int cows[maxn]={0};
int ans[maxn]={0};
void build(int left, int right, int loc){
tree[loc].left = left;
tree[loc].right = right;
tree[loc].len = right - left + 1;
if(left == right)
return;
int mid = (right + left)>>1;
build(left,mid,loc<<1);
build(mid+1,right,(loc<<1)+1);
}
int query(int num,int loc){
tree[loc].len -- ;
if(tree[loc].left == tree[loc].right){
return tree[loc].left;
}
if(tree[loc<<1].len < num){
return query(num-tree[loc<<1].len,(loc<<1)+1);
}
else{
return query(num,loc<<1);
}
}
int main(){
int n;
scanf("%d",&n);
for(int i=2;i<=n;++i){
scanf("%d",&cows[i]);
}
build(1,n,1);
for(int i=n;i>0;--i){
ans[i] = query(cows[i]+1,1);
}
for(int i=1;i<=n;++i){
printf("%d\n",ans[i]);
}
//system("pause");
return 0;
}