POJ2299 UltraQuickSort解题报告（树状数组+离散化模板）

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

　　　　　　题意：多组数据测试，给一个n和一个长度为n的序列，求此序列的逆序数；

　　　　 初学树状数组，理解其结构和功能（可查询前缀和），离散化一下，树状数组计数即可，详见代码及注释。

　　　　 外加离散化实现模拟。

#include<iostream>
#include<algorithm>
#include<cstring>
#define maxn 501000

using namespace std;

struct NUNO
{
    int w,pos;
}v[maxn];

int n;
int a[maxn];
int tr[maxn];

bool cmp(NUNO a,NUNO b)
{
    return a.w<b.w;
}

int low_bit(int x)
{
    return x&(-x);
}

void update(int i,int v)
{
    while(i<=n)
    {
        tr[i]+=v;
        i+=low_bit(i);
    }
}

long long get_sum(int x)
{
    long long s=0;
    while(x>=1)
    {
        s+=tr[x];
        x-=low_bit(x);
    }
    return s;
}


int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    //离散化
    /*

        例子：
    i    1 2 3        1 2 3
    v.w  9 8 7        7 8 9         i    1 2 3
  v.pos  1 2 3----->  3 2 1---->   a[i]  3 2 1

    */
    while(cin>>n,n){
    for(int i=1;i<=n;i++)
    {
        cin>>v[i].w;
        v[i].pos=i;
    }

    sort(v+1,v+n+1,cmp);

    for(int i=1;i<=n;i++)
    {
        a[v[i].pos]=i;
    }


    memset(tr,0,sizeof(tr));//无数据存在于数组 全标记为0

    long long ans=0;
    //大于a[i]的元素个数=总元素个数-小于等于a[i]的元素个数

    for(int i=1;i<=n;i++)
    {

        update(a[i],1);//将a[i]加入数组 将其标记为1 则包含a[i]的前缀和均+1

        ans+=(i-get_sum(a[i]));//get_sum()得到的是小于等于a[i]的元素数目

        //若再插入的a[i+1]>=a[i] 则a[i+1]的前缀和为1 说明小于等于a[i+1]的元素已经存在一个了
    }
    
    cout<<ans<<endl;
    }


    return 0;
}