树状数组求逆序数【模板】

树状数组模板转到 点击打开链接

可以用 c[i] 记录小于 i 的数字的个数。这个可以利用树状数组的特性，每添加一个数字i，向其父节点更新（父节点一定大于i），可以在log(n)的时间内完成更新。sum(i) 查询1~i区间内的小于i的个数。

计算逆序数的时候只需要用i-sum(i)就行了。

如果给出的数字范围过大，还需要进行离散化。离散化时需要记录其原本的位置。

给出一个经过离散化的模板：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<queue>
#include<math.h>
using namespace std;
#define MAXN 500005
int n,t[MAXN],c[MAXN];
struct node
{
    int x,i;
}p[MAXN];
bool cmp(node a,node b)
{
    return a.x < b.x;
}
int lowbit(int i)
{
    return i&(-i);
}
void add(int i,int x)
{
    while(i<=n)
    {
        c[i]+=x;
        i+=lowbit(i);
    }
}
int sum(int i)
{
    int res=0;
    while(i)
    {
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}
int main()
{
    while(scanf("%d",&n),n)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&p[i].x);
            p[i].i=i;
        }
        sort(p+1,p+n+1,cmp);
        for(int i=1;i<=n;i++)
            t[p[i].i]=i;
        long long ans=0;
        memset(c,0,sizeof c);
        for(int i=1;i<=n;i++)
        {
            add(t[i],1);
            ans+=i-sum(t[i]);
        }
        printf("%lld\n",ans);
    }
    return 0;
}