poj2182 Lost Cows

 1 #include<stdio.h>

 2 #define N 8010

 3 struct node{

 4     int l,r;

 5     int len;//len用来存放某一段数据的个数 

 6 }tree[N<<1];

 7 int s[N],result[N];

 8 void build(int l,int r,int i)

 9 {

10     tree[i].l=l;

11     tree[i].r=r;

12     tree[i].len=r-l+1; 

13     if(l==r) return;

14     int mid=(l+r)>>1;

15     build(l,mid,i<<1);

16     build(mid+1,r,(i<<1)+1);

17 }

18 int order(int s,int i)

19 {

20     tree[i].len--;//每进行一次取出操作，结点个数都要减一 

21     if(tree[i].l==tree[i].r) return tree[i].l;//树的l就是牛的编号 

22     if(s<=tree[i<<1].len) return order(s,i<<1);//假如要找的第s名的s<左子树的左边界，则在左子树上查找 

23     else return order(s-tree[i<<1].len,(i<<1)+1);//否则在右子树上查找，注意在右子树查找时，是查找第（s-tree[i<<1]）个节点的编号 

24 }

25 int main()

26 {

27     int i,n;

28     while(~scanf("%d",&n)){

29         build(1,n,1);

30         for(i=1;i<n;++i)

31             scanf("%d",&s[i]);

32         for(i=n-1;i>0;--i)

33             result[i]=order(s[i]+1,1);

34         result[0]=order(1,1);//最后一个就剩一个了，也就是第一个 

35         for(i=0;i<n;++i)

36             printf("%d\n",result[i]);

37     }

38     return 0;

39 }

搜索排在第n位的数是几时可用线段树实现，对于一个线段树中的所代表的线段[a,b]，结点中的数值纪录[a,b]中有多少人，记为len，所以对于一个在剩余数字队列中排在第n位，看这个人是在一个结点的左子树中还是右子树中，

判断的依据是用左子树的r和n比较，

如果r>=n说明在左子树形成的队列中有足够的数字使他排在第n位，于是递归找左子树。

如果r <n说明在左子树形成的队列中没有足够的数字使他排在第n位，所以他在右子树的第n-r位，于是递归在右子树中找;

当然每找到一个数字，要将所有的祖先结点信息改变，即len--;