poj 2182 树状数组

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

#include<iostream>
#include<algorithm>
using namespace std;
int mat[];
int main()
{
    int i,j,t,n;
    scanf("%d",&n);

         mat[]=;
         for(i=;i<n;i++)
         {
             scanf("%d",&t);
             mat[i]=t+;
             for(j=;j<i;j++)
               if(mat[j]>=mat[i])
                {  mat[j]++;}
         }
         for(i=;i<n;i++)
            printf("%d\n",mat[i]);

    return ;
}    

还是来看树状数组的解法。我们从后面往见面看,假设第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,,sizeof(C));
}
int Sum(int pos)
{
    int sum=;
    while(pos>)
    {
        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=,r=n,temp;
    while(l<r)
    {
        int mid=(l+r)>>;
        temp=mid-Sum(mid);
        if(temp<i)
            l=mid+;
        else
            r=mid;
    }
    return l;
}
int main()
{
    int i,j,a,b,c;
    while(scanf("%d",&n)!=EOF)
    {
        init();
        for(i=;i<=n;i++)
            scanf("%d",&Rank[i]);
        Rank[]=;
        for(i=n;i>=;i--)
        {
            int pos=getPos(Rank[i]+);
            ans[i]=pos;
            update(pos,);
        }
        for(i=;i<=n;i++)
            printf("%d\n",ans[i]);
    }
    return ;
}