POJ 2182 Lost Cows.(线段树）

~~~~

数据太水，暴力无压力，不过还是用线段树写了下，表现了楼主对线段树的无限热爱之情。

~~~~

题目连接：http://poj.org/problem?id=2182

大致题意;牛牛因XXXXXX原因乱序成一排，现已知每头牛前面有多少头牛比它的编号小（第一头牛前面没有当然就不列粗来了），求每头牛的编号。

思路：从后往前扫描，遇到a,则说明它是剩余序列的第a+1头牛。

~~~~

暴力代码：（157ms）#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 8888
using namespace std;

int q
,a
,ans
;
int main()
{
int n;
while(scanf("%d",&n)==1)
{
q[0]=0;a[0]=1;
for(int i=1;i<n;i++)
{
scanf("%d",&q[i]);
a[i]=i+1;
}
for(int i=n-1;i>=0;i--)
{
ans[i]=a[q[i]];
for(int j=q[i];j<n-1;j++)
a[j]=a[j+1];
}
for(int i=0;i<n;i++)
printf("%d\n",ans[i]);
}
return 0;
}
~~~~

线段树代码：（32ms）

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 8888
#define lson rt<<1,s,m
#define rson rt<<1|1,m+1,e
using namespace std;

int tre[N<<2];
void build(int rt,int s,int e)
{
tre[rt]=e-s+1; //用tre数组存每条线段间有多少个牛
if(s==e)
return ;
int m=(s+e)>>1;
build(lson);
build(rson);
}
int query(int val,int rt,int s,int e)
{
tre[rt]--; //每次query，牛数必减1，因为下一次询问总是从剩余的序列中。
if(s==e)
return s; //~~
int m=(s+e)>>1;
if(val<=tre[rt<<1]) return query(val,lson);
else return query(val-tre[rt<<1],rson);
}

int q
,ans
;
int main()
{
int n;
while(scanf("%d",&n)==1)
{
build(1,1,n);
q[0]=0; //~~
for(int i=1;i<n;i++)
scanf("%d",&q[i]);
for(int i=n-1;i>=0;i--)
ans[i]=query(q[i]+1,1,1,n);
for(int i=0;i<n;i++)
printf("%d\n",ans[i]);
}
return 0;
}