AcWing 788. 逆序对的数量（归并排序||树状数组）

整理的算法模板：ACM算法模板总结（分类详细版）

 

给定一个长度为n的整数数列，请你计算数列中的逆序对的数量。

逆序对的定义如下：对于数列的第 i 个和第 j 个元素，如果满足 i < j 且 a[i] > a[j]，则其为一个逆序对；否则不是。

输入格式

第一行包含整数n，表示数列的长度。

第二行包含 n 个整数，表示整个数列。

输出格式

输出一个整数，表示逆序对的个数。

数据范围

1≤n≤1000001≤n≤100000

输入样例：

6
2 3 4 5 6 1

输出样例：

5

 求逆序对当然还是用归并排序好一点，比树状数组快一些，比如这道题，归并排序速度大概150ms，比树状数组快一倍；

归并排序：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005;
ll a[N],ans,n,res[N];
void merge_sort(int l,int r)
{
    if(l>=r) return ;
    int mid=l + r >>1;
    merge_sort(l,mid),merge_sort(mid+1,r);
    int i=l,j=mid+1,k=0;
    while(i<=mid&&j<=r)
    {
        if(a[i]<=a[j]) res[k++]=a[i++];
        else res[k++]=a[j++],ans+=mid-i+1;
    }
    while(i<=mid) res[k++]=a[i++];
    while(j<=r) res[k++]=a[j++];
    for(int i=l,j=0;i<=r;i++,j++) a[i]=res[j];
}
int main()
{
    cin >>n;
    for(int i=0;i<n;i++) cin >>a[i];
    merge_sort(0,n-1);
    cout <<ans<<endl;
}

 

树状数组求逆序对

（先离散化处理数据，注意是保序离散化）

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005;
int tr[N],n,a[N];
vector<int> alls; // 存储所有待离散化的值
 
// 二分求出x对应的离散化的值
int find(int x) // 找到第一个大于等于x的位置
{
    int l = 0, r = alls.size() - 1;
    while (l < r)
    {
        int mid = l + r >> 1;
        if (alls[mid] >= x) r = mid;
        else l = mid + 1;
    }
    return r + 1; // 映射到1, 2, ...n
}
int lowbit(int x)
{
	return x&-x;
}
int modify(int x,int c)//修改树状数组x位置的值
{
	for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}
int query(int x)//查询区间1~x的区间和；
{
	int res=0;
	for(int i=x;i>=1;i-=lowbit(i)) res+=tr[i];
	return res; 
}
int main()
{
    cin >>n;
    long long ans=0;
    for(int i=1;i<=n;i++) cin >>a[i],alls.push_back(a[i]);
    sort(alls.begin(), alls.end()); // 将所有值排序
    alls.erase(unique(alls.begin(), alls.end()), alls.end());   // 去掉重复元素
    for(int i=1;i<=n;i++)
    {
        a[i]=find(a[i]);
        ans+=query(n)-query(a[i]);
        modify(a[i],1);
    }
    cout <<ans<<endl;
    
}