poj 2182

有一群 牛 排成 一行 (编号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]);
	}
}

方法三。线段树,有待学习。。





















你可能感兴趣的:(poj 2182)