poj 2182 树状数组

这题对于O(n^2)的算法有很多,我这随便贴一个烂的,跑了375ms。

#include<iostream>
#include<algorithm>
using namespace std;
int mat[8008];
int main()
{
    int i,j,t,n;
    scanf("%d",&n);
    
         mat[0]=1;
         for(i=1;i<n;i++) 
         {
             scanf("%d",&t);
             mat[i]=t+1;                    
             for(j=0;j<i;j++)                       
               if(mat[j]>=mat[i])
                {  mat[j]++;}
         }
         for(i=0;i<n;i++)
            printf("%d\n",mat[i]);
    
    return 0;
}    
View Code


还是来看树状数组的解法。我们从后面往见面看,假设第i个牛前面有Rank[i]个牛标号比他小,由此很容易得出最后一个牛的编号一定是Rank[n]+1。我们将最后一个牛去掉后,剩下n-1牛时,对于第n-1个牛的道理是一样的,只不过不同的是Rank[n-1]+1之前有多少个牛已经不存在了,即已经确定位置。我们就用树状数组保存被去掉的牛个数,C[i]表示标号i之前已经有多少个牛确定位置。那么我们每次就是要找某个数k,使得k-C[k]==Rank[i]+1。这就是第i个牛的确定位置。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define Maxn 8010
#define inf 0x7fffffff
#define lowbit(x) (x&(-x))
using namespace std;

int C[Maxn],n,Rank[Maxn],ans[Maxn];
void init()
{
    memset(C,0,sizeof(C));
}
int Sum(int pos)
{
    int sum=0;
    while(pos>0)
    {
        sum+=C[pos];
        pos-=lowbit(pos);
    }
    return sum;
}
void update(int pos,int num)
{
    while(pos<=n)
    {
        C[pos]+=num;
        pos+=lowbit(pos);
    }
}
int getPos(int i)
{
    int l=1,r=n,temp;
    while(l<r)
    {
        int mid=(l+r)>>1;
        temp=mid-Sum(mid);
        if(temp<i)
            l=mid+1;
        else
            r=mid;
    }
    return l;
}
int main()
{
    int i,j,a,b,c;
    while(scanf("%d",&n)!=EOF)
    {
        init();
        for(i=2;i<=n;i++)
            scanf("%d",&Rank[i]);
        Rank[1]=0;
        for(i=n;i>=1;i--)
        {
            int pos=getPos(Rank[i]+1);
            ans[i]=pos;
            update(pos,1);
        }
        for(i=1;i<=n;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}

 

posted @ 2013-07-30 15:46  fangguo  阅读(188)  评论(0编辑  收藏  举报