[POJ 2182] Lost Cows · 树状数组

着手准备pku夏令营所以得把英文题好好刷一刷。

写树状数组只是觉得比较好写

一开始机房里的小伙伴想到了逆推的方法，但没有想出来怎么确定值，然后发现可以用树状数组来搞。

从后往前搞，第i个位置上原本的值是ans[i]，那么之前比ans[i]小的数就是a[i] ，之后比ans[i]小的数就是sum(ans[i])，所以ans[i] = a[i] + sum( ans[i] ) + 1 ，其中a[i]是读入的数据，sum( ans[i] ) 用树状数组求，然后我们对ans[i]二分答案，如果p + a[i] + sum( p ) < p，说明ans[i] > p，反过来也同理。

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <iostream>
using namespace std;
#define lowbit(x) ((x) & (-x))

const int N = 8005 ;
int a[N],n,s[N],ans[N],x;

int sum(int t) {
	int p = 0;
	for (int i = t ; i ; i -= lowbit(i) ) p += s[i] ;
	return p ;
}

void add(int t) {
	for (int i = t ; i <= n ; i += lowbit(i) ) s[i] ++ ;
	return ;
}

int binary(int t) {
	int l = 1 , r = n ,mid ;
	while ( l < r ) {
		mid = ( l + r ) >> 1 ;
		int p = sum( mid ) + 1 + t ;
		if ( p <= mid ) r = mid ;
			else l = mid + 1 ;
	}
	return l ;
}

int main() {
	cin >> n ; a[1] = 0 ;
	for (int i = 2 ; i <= n ; i ++ ) cin >> a[i] ; 
	for (int i = n ; i ; i -- ) {
		x = binary( a[i] ) ; 
		add( x ) ;
		ans[i] = x ;
	}
	for (int i = 1 ; i <= n ; i ++ )
		cout << ans[i] << endl ;
	return 0 ;
}