有一群 牛 排成 一行 (编号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]); } }
方法三。线段树,有待学习。。