树状数组 (hdu2689)离散化

求一组数的逆序数的个数

对于50%的数据,n≤2500

对于100%的数据,n≤40000。

n<=40000,但是 每次输入的数不一定<40000,有可能会很大,树状数组装不下,此时需要将

数据离散化

************************************************************************************************************************************

例如 一个数组

9    12    3    24    6    8     最大的数为24   树状数组需要开到24    进行离散化

第一步:输入+编号

先建立一个结构体      val存数值,no存编号

struct node
{
    int val;
    int no;
} c[40005];

此时  数组变为      9    12    3    24    6    8

                             1     2      3    4      5    6

第二步:按照val大小进行排序

         数组变为      3     6     8     9    12    24

                              3     5     6     1    2      4

第三步:将此数组以no的顺序映射到一个新的公差为1的数组中

                              3     6     8     9    12    24

                              3     5     6     1    2      4

                              1      2    3     4    5      6

                            a[3]  a[5] a[6] a[1] a[2]  a[4]

                                4      5      1     6     2     3             (4为第四大的数  其他的一致)

                              a[1]  a[2] a[3] a[4] a[5]  a[6]

离散化完成   现在只需要将树状数组开到5

**********************************************************************************************************************************

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int tree[40005];
int n;
int a[40005];
struct node
{
    int val;
    int no;
} c[40005];
bool cmp(node a,node b)
{
    return a.val<b.val;
}
int lowbit(int i)
{
    return i&(-i);
}
void update(int i,int x)
{
    while(i<=40005)
    {
        tree[i]+=x;
        i+=lowbit(i);
    }
}
int query(int i)
{
    int sum=0;
    while(i>0)
    {
        sum+=tree[i];
        i-=lowbit(i);
    }
    return sum;
}
int main()
{
    int sum=0;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&c[i].val);
        c[i].no=i;
    }
    sort(c+1,c+1+n,cmp);
    for(int i=1; i<=n; i++)
        a[c[i].no]=i;
    for(int i=1;i<=n;i++)
        {
            update(a[i],1);
            sum+=i-query(a[i]);
        }
    printf("%d\n",sum);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xqx1343002589/article/details/81265436