POJ 2299 Ultra-QuickSort(树状数组求逆序数)

http://poj.org/problem?id=2299

用树状数组求逆序数和用线段树求逆序数方法类似。
输入9 1 0 5 4,那么C[i]树状数组的建立是在,
下标 0 1 2 3 4 5 6 7 8 9
数组 1 1 0 0 1 1 0 0 0 1。
因此二者都需要离散化。
然后大概思路就是边更新边求值。


#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long LL;

const int N = 500000 + 100;
long long a[N],b[N];
long long n;
long long tr[N];
long long lowbit(long long x)
{
    return x & (-x);
}
void update(long long pos,long long val)
{
    while(pos<=n)
    {
        tr[pos] += val;
        pos += lowbit(pos);
    }
    return;
}
long long query(long long x)
{
    long long sum = 0;
    while(x)
    {
        sum += tr[x];
        x -= lowbit(x);
    }
    return sum;
}
long long solve()
{
    long long res = 0;
    for(long long i = n;i>=1;i--)
    {
        long long x = lower_bound(b+1,b+1+n,a[i])-(b+1) + 1;
        res += query(x-1);
        update(x,1);
    }
    return res;
}
int main()
{
    while(~scanf("%lld",&n) && n) 
    {
        memset(tr,0,sizeof(tr));
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        for(long long i=1;i<=n;i++)
            scanf("%lld",&a[i]),b[i] = a[i];
        sort(b+1,b+1+n);
        printf("%lld\n",solve());
    }
    return 0;
}

你可能感兴趣的:(*Data,Structure)