有一群 牛 排成 一行 (编号1-N),现在 给出 每一头 牛 前面 有几头 牛 的 编号 比 该 牛 编号小 ,要 求 每个 位置上 牛的编号
如 给出 1 2 1 0 (用 数组 sm存储),对应 的 结果 为 2 4 5 3 1 (记录 在 bd数组中),就是 说 4 的前面 有一个 编号 比他小 ,也就是 2 ,5 的前面 有两个 (2 和 4)。。。
方法 一,暴力枚举 ,从 后面 开始 ,最后 一个 即sm[N]+1 就是 该牛编号,标记 vis[]=1;对于 倒数第二头 牛 ,从 vis[1...N]找j,如果 vis没标记过则 small++;直到small==sm[i]&&该vis没标记过,则该位置 为 牛 i 的 编号。。
#include<cstdio>
#include<cstring>
#define for if(0);else for
const int maxn=10005;
int brand[maxn];
int small[maxn];
int vis[maxn];
int main(){
int n;
while(scanf("%d",&n)==1){
small[1]=0;
for(int i=2;i<=n;i++)scanf("%d",&small[i]);
memset(vis,0,sizeof(vis));
memset(brand,0,sizeof(brand));
for(int i=n;i>=1;i--){
//brand[i]=small[i]+1;
int smaller=0,j;
for(j=1;j<=n;j++){
if(smaller==small[i]&&vis[j]==0)break;
if(vis[j]==0)smaller++;
}
brand[i]=j;
//printf("%d\n",smaller);
//printf("%d\n",brand[i]);
vis[j]=1;
}
for(int i=1;i<=n;i++)printf("%d\n",brand[i]);
}
}
方法二,树状数组。。
一开始没想到,看了一些大神的报告,总算是 理解了。用 树状数组来 维护 前 k 个值的 和,用到 的数组 记为 c[],初始化为 1,表示 该位置 可以 站牛,对应的 如果 为 0就表示 该位置已经有 牛了。从后往前算,和 上一方法一样,找出最后 一头牛编号 ,那么就可以 找 倒数第二头了,每个位置 上的 牛编号 : 前 k 项和 ==sm[]+1. 并且 k 位置 值为 1,也就是说 该位置 可以站牛,用 二分 法 从 1到 N中 求 得 k 即可。。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define for if(0);else for
const int maxn=10005;
int bd[maxn];
int sm[maxn];
int c[maxn*3];
int n;
int lowbit(int x){
return x&(-x);
}
void update(int i,int v){
while(i<=maxn){
c[i]+=v;
i+=lowbit(i);
}
}
int getsum(int i){
int sum=0;
while(i>0){
sum+=c[i];
i-=lowbit(i);
}
return sum;
}
int bsearch(int v){
int low=1;
int high=n;
while(low<=high){
int mid=(low+high)>>1;
if(getsum(mid)>=v)high=mid-1;
else low=mid+1;
}
return low;
}
int main(){
while(scanf("%d",&n)==1){
for(int i=2;i<=n;i++)scanf("%d",&sm[i]);
for(int i=1;i<=n;i++)update(i,1);
for(int i=n;i>=1;i--){
bd[i]=bsearch(sm[i]+1);
update(bd[i],-1);
}
for(int i=1;i<=n;i++)printf("%d\n",bd[i]);
}
}
方法三。线段树,有待学习。。