树状数组的应用（SOJ2559）

(2012-07-23 14:04:18)

SOJ2559：http://cstest.scu.edu.cn/soj/problem.action?id=2559

这道题是说对于给定的一个数组s求s[i]在前i个元素中的排名（从大到小）。

其实很简单，对于s[i]只需找出前i个元素中

题中的数据量为45000，对于每个元素的值没有具体的要求，只是为int型。

首先对数组s进行归一化（我自己的命名，很多人说是离散化，我觉得归一化更为恰当）处理，将数组中的元素值映射到1-45000上。

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
struct node
{
    int num;
    int rank;
}s[45005];
int t[45005];
int c[45005];
int result[45005];
int n;/
int lowbit(int x)
{
    return x&(x^(x-1));
}
void mod(int i)
{
    while(i<=n)
    {
        c[i]++;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int SUM=0;
    while(i)
    {
        SUM+=c[i];
        i-=lowbit(i);
    }
    return SUM;
}
bool cmp(node a,node b)
{
    return a.num<b.num;
}
int main()
{
    int i;
    while(scanf("%d",&n)==1)
    {
        memset(c,0,sizeof(c));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&s[i].num);
            s[i].rank=i;
        }
        sort(s+1,s+n+1,cmp);
        for(i=1;i<=n;i++)
            t[s[i].rank]=i;
        for(i=1;i<=n;i++)
        {
            result[i]=i-sum(t[i]);
            mod(t[i]);
        }
        for(i=1;i<=n;i++)
            printf("%d\n",result[i]);
    }
    return 0;
}