天天看點

poj——2299樹狀數組求逆序數

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=500005;
struct Node
{
    int pos,val;
}a[N];
int c[N],num[N],n;
bool cmp(Node n1,Node n2)
{
    return n1.val<n2.val;
}

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

void add(int x,int val)
{
    while(x<=n)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}

int query(int x)
{
    int sum=0;
    while(x)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    while(~scanf("%d",&n)&&n)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i].val);
            a[i].pos=i;
        }
        sort(a+1,a+n+1,cmp);
        for(int i=1;i<=n;i++)
        {
            num[a[i].pos]=i;//原先放在pos位置的元素,排序後放在i位置。
        }
        long long ans=0;
        for(int i=1;i<=n;i++)
        {
            add(num[i],1);//依次插入,從後往前找比他小的數
            ans+=i-query(num[i]);//
            //printf("%d\n",ans);
        }
        cout<<ans<<endl;
    }
    return 0;
}