POJ 2182 ：Lost Cows

题目链接：http://acm.pku.edu.cn/JudgeOnline/problem?id=2182

 

题意：FJ有n头牛，编号为1~n，它们并没有按照编号的顺序排好队列。现在，FJ只知道每一个牛前面有多少只牛的编号比它大。问你能不能判断出所有牛的编号。

 

思路：线段树。关键：每次最后一只牛的编号是可以确定的，即为a[i]+1，将其编号从所有牛中删除，则倒数第二只牛的编号又可以确定为a[i]+1，依此类推

 

AC代码：

#include<cstdio> #include<cstring> #include<cmath> #include<algorithm> using namespace std; const int maxn =8010; struct node{ int l,r; int cnt; }; node Tree[3*maxn]; int ans[maxn]; int a[maxn]; int Rank[maxn]; int n; void BuildTree(int root,int l,int r)//建树 { Tree[root].l=l; Tree[root].r=r; Tree[root].cnt=r-l+1; if(l<r) { BuildTree(2*root,l,(l+r)>>1); BuildTree(2*root+1,((l+r)>>1)+1,r); } } void Delete(int root,int c)//维护，关键一点：所求值为当前区间的左起第c个元素。对访问到的区间的len进行-1 { if(Tree[root].l==Tree[root].r) { ans[n]=Tree[root].l; Tree[root].cnt=0; return ; } if(c<=Tree[2*root].cnt) { Delete(2*root,c);//情况1:左区间的数字足够，则查询左区间起的第c个元素 } else Delete(2*root+1,c-Tree[2*root].cnt);//情况2：左子区间的数字个数不够，则查询右子区间的左起第c - node[root*2].len个元素 Tree[root].cnt--; } int main() { //freopen("input.in","r",stdin); int N; int num; int i; while(scanf("%d",&N)!=EOF) { BuildTree(1,1,N); a[1]=0; for(i=2;i<=N;i++) scanf("%d",&a[i]); for(n=N;n>0;n--) Delete(1,a[n]+1); for(i=1;i<=N;i++) printf("%d/n",ans[i]); } return 0; }