树状数组求逆序对 + 离散化(poj2299)

转载注明出处
https://www.cnblogs.com/xiongmao-cpp/p/5043340.html

今天做了一个树状数组求逆序对的题，需要离散化，看了博客明白了为什么要进行离散化，原因是树状数组中的C[MAXN]数组其实相当于一个哈希操作，如果所给数组中存在值特别大的元素，数组就需要离散化

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
struct A
{
    int i;
    int v;
};
const int MAXN = 5e5 + 10;
A a[MAXN];
int c[MAXN];
typedef long long LL;
int lowbit(int i)
{
    return i & (-i);
}
int n;
void add(int i,int v)
{
    while(i <= n)
    {
        c[i] += v;
        i += lowbit(i);
    }
    return ;
}
int query(int i)
{
    int ans = 0;
    while(i)
    {
        ans += c[i];
        i -= lowbit(i);
    }
    return ans;
}
bool cmp(A a,A b)
{
    return a.v < b.v;
}
int main()
{
    ios::sync_with_stdio(false);
    while(cin >> n && n)
    {
        memset(c,0,sizeof(c));
        for(int i = 1; i <= n; ++i)
        {
            cin >> a[i].v;
            a[i].i = i;
        }
        sort(a + 1,a + n + 1,cmp);
        LL ans = 0;
        for(int i = 1; i <= n; ++i)
        {
            add(a[i].i,1);
            ans += i - query(a[i].i);
        }
        cout << ans << endl;
    }
    return 0;
}