[Usaco2003 Open]Lost Cows(线段树)

题目链接：http://begin.lydsy.com/JudgeOnline/problem.php?id=2875

类似于poj上的一道题，具体记不得了。

算是线段树的一道基础题吧，单点更新问题。

sum[rt]记录前面所有的空闲位置的数目

sum[rt]=sum[rt<<1]+sum[rt<<1|1];

从后往前进行插入，每次插入之后记录插入的位置，并做一个标记，第一个数应该为最终未标记的那个数，然后依次输出即可。

注意一下更新时的判断条件即可

if(v<sum[rt<<1]) Update(lson,id,v);
else Update(rson,id,v-sum[rt<<1]);

具体代码如下：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;

#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r

typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn=1e4+10;
int T,n;
int v[maxn];
int sum[maxn<<2];
int ans[maxn];
bool flag[maxn];

void init(){
	memset(ans,-1,sizeof(ans));
	memset(flag,false,sizeof(flag));
}

void PushUp(int rt){
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void build(int rt,int l,int r){
	sum[rt]=r-l+1;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(lson);
	build(rson);
}

void Update(int rt,int l,int r,int id,int v){
	if(l==r){
		sum[rt]=0;
		ans[id]=l;
		flag[l]=true;
		return ;
	}
	int mid=(l+r)>>1;
	if(v<sum[rt<<1]) Update(lson,id,v);
	else Update(rson,id,v-sum[rt<<1]);
	PushUp(rt);
}

int main(){
#ifndef ONLINE_JUDGE
	freopen("test.in","r",stdin);
	freopen("test.out","w",stdout);
#endif
	while(~scanf("%d",&n)){
		for(int i=0;i<n-1;i++){
			scanf("%d",v+i);
		}
		init();
		build(1,1,n);
		for(int i=n-2;i>=0;i--){
			Update(1,1,n,i,v[i]);
		}
		int index=-1;
		for(int i=1;i<=n;i++){
			if(!flag[i]){
				index=i;
				break;
			}
		}
		printf("%d\n",index);
		for(int i=0;i<n-1;i++){
			printf("%d\n",ans[i]);
		}
	}
	return 0;
}