有n头奶牛,已知它们的身高为 1~n 且各不相同,但不知道每头奶牛的具体身高。现在这n头奶牛站成一列,已知第i头牛前面有Ai头牛比它低,求每头奶牛的身高。
输入格式
第1行:输入整数n。
第2…n行:每行输入一个整数Ai,第i行表示第i头牛前面有Ai头牛比它低。
(注意:因为第1头牛前面没有牛,所以并没有将它列出)
输出格式
输出包含n行,每行输出一个整数表示牛的身高。
第i行输出第i头牛的身高。
数据范围
1≤n≤105
输入样例:
5
1
2
1
0
输出样例:
2
4
5
3
1
先说一下树状数组,树状数组是个很神奇的数据结构,定义tr[x]为区间长度为lowbit(x),末尾时x的区间和,就是[x-lowbit(x)+1,x]的区间和。
lowbit(x)=x&-x;
int lowbit(int x)
{
return x&-x;//返回二进制下最后一位1所对应的十进制,假设最后一位1是在第i位,就返回2^i
}
tr[x]是[x-lowbit(x)+1,x]的和,那么如果想求[1,n]的和该怎么求,tr[x-lowbit(x)]是不是刚好与tr[x]无缝相连,那么一直这项拼接下去直到区间起点为1就可以了(可以证明这样是一定存在的)
int sum(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i)) res+=tr[i];
return res;
}
如果还想修改某个点的值那该怎么做呢,假设要修改点x位置的值,只需要找一下有那么些 tr