线段树入门 POJ 2182 Lots Cows

看别人的思路：

从后往前推，排在最后一头牛比他编号小的数量有x头牛，那么最后一头牛的编号必然为x+1。

按照这种思路下去，我们把1-n编号的牛排成一排。把x+1这头牛从里面删除。

假如倒数第二头牛在他前面比他编号小的数量有y头牛，那么就在这1-n删掉了求出来的牛后剩下的第y+1头牛。

方法一：直接搜索（n*n）

p[i]表示第i个应该放的cow的id。

f[i]＝0表示此牛还未被删除，f[i]=1表示此牛已经被删除。

#include <iostream> #include <cstdio> #include <cstring> using namespace std; int n; int a[8010]; int p[8010]; int f[8010]; int main() { scanf("%d",&n); int i =1; a[1]=1; for(i=2;i<=n;i++) { scanf("%d",&a[i]); } a[1]=0; for(i=n;i>0;i--) { int k = 0; int j =0; for(j=1;j<=n;j++) if(f[j]==0) { k++; if(k==a[i]+1) break; } p[i]=j; f[j]=1;//dele k-th cow } for(i=1;i<=n;i++) printf("%d/n",p[i]); }

方法二：线段树（n*lgn） 

搜索排在第n位的数是几时可用线段树实现 ，对于一个线段树中的所代表的线段[a,b]，结点中的数值纪录[a,b]中还有多少人没有被去掉，记为len，所以对于一个在剩余数字队列中排在第n位，看这个人是在一个结点的左子树中还是右子树中。

#include <iostream> #include <cstdio> #include <cstring> using namespace std; struct node { int l,r; int cnt;//在[l,r]区间中的个数 }tree[16*1024]; int save[8010]; int a[8010]; void build(int l,int r,int u) { tree[u].l=l; tree[u].r=r; tree[u].cnt = r-l+1; if(l<r) { build(l,(l+r)/2,2*u); build((l+r)/2+1,r,2*u+1); } } void del(int c,int root) { if(tree[root].l==tree[root].r) { save[++save[0]]=tree[root].l; tree[root].cnt=0; return; } if(c<=tree[2*root].cnt) del(c,2*root); else del(c-tree[2*root].cnt,2*root+1); tree[root].cnt--; } int main() { int n ; scanf("%d",&n); int i =0; a[1]=0; for(i=2;i<=n;i++) scanf("%d",&a[i]); build(1,n,1); for(i=n;i>=1;i--) del(a[i]+1,1); for(i=save[0];i>=1;i--) printf("%d/n",save[i]); }