NKOJ 3709 走丢的奶牛

问题描述

约翰有n头奶牛，这些奶牛长得很相似，约翰经常分不清谁是谁，于是约翰给他们编号1到n，以此来区分每头奶牛。 今天奶牛们想作弄一下约翰，n头奶牛乱序排成一排。 约翰想要分清每个位置对应奶牛的编号。约翰从左起第2头奶牛开始一直到第n头奶牛，依次问每个奶牛一个问题“你左边有多少个编号比你小的奶牛”，奶牛们都会如实回答约翰的问题。 问，你能否根据奶牛的回答，帮助约翰推出这n头牛对应的编号？

---

输入格式

第一行，一个整数n 接下来n-1行，每行一个整数，分别表示左起第2到第n头奶牛的回答。

---

输出格式

n行，每行一个整数，对应从左往右每头奶牛的编号

---

样例输入

5 1 2 1 0

---

样例输出

2 4 5 3 1

---

数据范围

1<=N<=50000

---

树状数组，把数组装好后弄一个二分就行了 为什么会想到二分呢，我也不是很清楚== 看到这种编号自然而然的一种感觉

#include<iostream> #include<algorithm> using namespace std; int n; int a[50005],ans[50005],c[50005]; int lowbit(int x) { return x&-x; } int getsum(int x) { int sum=0; for(int i=x;i>0;i-=lowbit(i))sum+=c[i]; return sum; } void modify(int x,int y) { for(int i=x;i<=n;i+=lowbit(i))c[i]+=y; } int main() { int mid,Left,Right; scanf("%d",&n); for(int i=2;i<=n;i++)scanf("%d",&a[i]); for(int i=n;i>0;i--) //二分 { Left=a[i]+1; Right=a[i]+1+n-i; while(Left<=Right) { mid=(Right+Left)/2; if(getsum(mid)+a[i]<=mid-1)Right=mid-1; else Left=mid+1; } ans[i]=Left; modify(Left,1); } for(int i=1;i<=n;i++)cout<<ans[i]<<endl; }