分治-Acwing-逆序对的数量

分治-Acwing-逆序对的数量

题目：

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

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

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

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

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

数据范围
1≤n≤100000
输入样例：
6
2 3 4 5 6 1
输出样例：
5

题解：

$首先，最暴力的做法，两重循环i,j遍历数组a,对每一个a[i],遍历出剩下所有的a[j]<a[i]，O(n^2)。$

$$\begin{gathered} 考虑分治法优化，对区间[l,r]之间的逆序对的数量，可以分为三个部分： \\ ①、i,j在区间[l,mid]，②、i,j在区间[mid+1,r]之间，③、i在[l,mid]中，j在[mid+1,r]中。 \end{gathered}$$

$$\begin{gathered} 对于第③部分，由于a[i],a[j]分居两边，因此将两边区间有序与否不会影响逆序对数量的统计， \\ 所以可以将两边区间排序后再统计，这样可以将第三部分的复杂度降到O(lo{g}_{2}n)。 \end{gathered}$$

$而①、②两部分，可以通过分治的方法，转化到第③部分。$

$这其实是一个归并排序的过程，边排序边计算。整个复杂度O(nlo{g}_{2}n)。$

$如上图，若a[i]>a[j]，那么从下标i到mid的所有元素都大于a[j]，数量为mid-i+1。$

---

归并排序模板：

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=1e5+10;
int tmp[N];
void merge_sort(int q[],int l,int r)
{
    if(l>=r) return;

    int mid=l+r>>1;

    merge_sort(q,l,mid),merge_sort(q,mid+1,r);

    int i=l,j=mid+1;
    int k=0;

    while(i<=mid&&j<=r)
        if(q[i]<=q[j]) tmp[k++]=q[i++];
        else tmp[k++]=q[j++];
    while(i<=mid) tmp[k++]=q[i++];
    while(j<=r) tmp[k++]=q[j++];

    for(int i=l,j=0;i<=r;i++,j++) q[i]=tmp[j];

}
int main()
{
    int n,a[100010];
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    merge_sort(a,0,n-1);
    for(int i=0;i<n;i++) printf("%d ",a[i]);
    return 0;
}

---

本题代码：

注意：
$由于数据范围是10^5，当数组是逆序时，最大答案达到\frac{(n-1)(n-2)}{2},大约在5\times 10^9爆了int，故答案用ll存。$

#include<iostream>
#include<algorithm>
#include<cstdio>
#define ll long long
using namespace std;
const int N=1e5+10;
int tmp[N];
ll ans;
void merge_sort(int q[],int l,int r)
{
    if(l>=r) return;

    int mid=l+r>>1;

    merge_sort(q,l,mid),merge_sort(q,mid+1,r);//左半区间+右半区间

    int i=l,j=mid+1;
    int k=0;

    while(i<=mid&&j<=r)
        if(q[i]<=q[j]) tmp[k++]=q[i++];
        else
        {
            tmp[k++]=q[j++];
            ans+=mid-i+1;//统计答案
        }
    while(i<=mid) tmp[k++]=q[i++];
    while(j<=r) tmp[k++]=q[j++];

    for(int i=l,j=0;i<=r;i++,j++) q[i]=tmp[j];

}
int main()
{
    int n,a[100010];
    scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    merge_sort(a,0,n-1);
    cout<<ans<<endl;
    //for(int i=0;i<n;i++) printf("%d ",a[i]);
    return 0;
}