poj2299_树状数组II

这个题做过了，今天是树状数组的另一个版本。

分析：

例如 9 1 0 5 4

一般数据小的话，我们求每个数字前面小于等于它的个数，只需要离散到大于等于1的范围内。

9： update(10),sum(10) --->1

1:   update(2),sum(2)--->1

0:   update(1),sum(0)--->1

5:   update(6),sum(6)--->3

4:   update(5),sum(5)--->3

但是这个题数据范围很大，因此必须得压缩一下

注意到 9在序列中第5大，1第2大，。。。。

因此，将 9 1 0 5 4替换成5 2 1 4 3做树状数组是不变的

然后对应的5-1;2-1;1-1;4-3;3-3;求和就可得逆序数。

代码：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <memory.h>
using namespace std;
const int maxnum=500000;
struct node
{
    int digit;
    int num;
}array[maxnum],temp[maxnum];
int tree[maxnum];
int n;

bool cmp(struct node a,struct node b)
{
    return a.digit<b.digit;
}

void update(int index,int add)
{
    while(index<=n)  //  1.是<=n
    {
        tree[index]+=add;
        index+=((-index)&index);
    }
}

int getsum(int index)
{
    int sum=0;
    while(index>0)
    {
        sum+=tree[index];
        index-=((-index)&index);
    }
    return sum;
}


int main()
{
    int i;
    long long sum;
    while(scanf("%d",&n) && n!=0)
    {
        for(i=1;i<=n;i++)
        {
            scanf("%d",&array[i].digit);
            temp[i].digit=array[i].digit;
            temp[i].num=i;
        }
        sort(temp+1,temp+n+1,cmp);
        for(i=1;i<=n;i++)
            array[temp[i].num].num=i;
        memset(tree,0,sizeof(tree)); //初始化树状数组
        sum=0;
        for(i=1;i<=n;i++)
        {
            update(array[i].num,1);
            sum+=(array[i].num-getsum(array[i].num));
        }
        printf("%I64d\n",sum);
        //cout<<sum<<endl;
    }
    return 0;
}