【题意】
有n头奶牛,已知它们的身高为 1~n 且各不相同,但不知道每头奶牛的具体身高。
现在这n头奶牛站成一列,已知第i头牛前面有Ai头牛比它低,求每头奶牛的身高。
一道裸题.很明显从后往前扫,然后找到未选的数中第a[i]+1大即可.
方法1:权值线段树
在权值线段树上二分,并顺便删除.
需要内存.时间复杂度
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lc (x<<1)
#define rc (x<<1|1)
using namespace std;
const int N=1e5+10;
int s[N<<2];
void bt(int x,int l,int r) {
s[x]=r-l+1;
if(l==r)return;
int mid=(l+r)>>1;
bt(lc,l,mid);
bt(rc,mid+1,r);
}
int ans;
void find(int x,int l,int r,int k) {
s[x]--;
if(l==r){ans=l;return ;}
int mid=(l+r)>>1;
if(k<=s[lc])find(lc,l,mid,k);
else find(rc,mid+1,r,k-s[lc]);
}
int n,a[N];
int main() {
scanf("%d",&n);bt(1,1,n);
for(int i=2;i<=n;i++)scanf("%d",&a[i]);
for(int i=n;i ;i--) {
find(1,1,n,a[i]+1);
a[i]=ans;
}
for(int i=1;i<=n;i++)printf("%d
",a[i]);
return 0;
}
方法2:树状数组+倍增
设b维护前缀可选位置的个数.(树状数组)
从后往前扫,对于每个位置,步骤如下:
- 初始化
- 从枚举,若
. - (记录答案),
步骤2非常巧妙的一点是由于指数从大到小枚举
这个算法只需要内存.时间复杂度(常数都比线段树小)
代码:
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e5+10;
int n,t,a[N],b[N];
void del(int x) {for( ;x<=n;x+=x&-x)b[x]--;}
int main() {
scanf("%d",&n);
t=log2(n);b[1]=1;
for(int i=2;i<=n;i++)scanf("%d",&a[i]),b[i]=i&-i;
for(int i=n;i ;i--) {
int ans=0,sum=0;
for(int j=t;j>=0;j--)
if(ans+(1<<j)<=n&&sum+b[ans+(1<<j)]<=a[i])
ans+=1<<j,sum+=b[ans];
del(a[i]=ans+1);
}
for(int i=1;i<=n;i++)printf("%d
",a[i]);
return 0;
}
方法3:树状数组(线段树)+二分
时间复杂度
枚举位置,判断.
这种方法太过暴力,所以我就没有打了.