/*
使用归并排序解决逆序对问题
步骤:
1.同理归并排序把整个数组分为左右两部分,再分别排序
2.在合并到tem临时数组的时候把逆序对的数量加上
逆序对的分布情况:
1):构成逆序对的两个数都在左边的数量:merge_sort(l, mid)
2):构成逆序对的两个数都在右边的数量:merge_sort(mid + 1, r)
3):构成逆序对的两个数一左一右的数量:在合并到tem临时数组时,q[i] > q[j] 就加
将三部分的逆序对数量加起来就是answer
*/
#include <iostream>
using namespace std;
//1≤n≤100000,当最大数量的数组全部完全逆序时
//逆序对的数量为5*10^9,大于int最大值2*10^9
//所以用long long来保存结果,防止溢出
typedef long long LL;
const int N = 1e5;
int n;
int q[N], tem[N];
LL merge_sort(int l, int r)
{
if(l >= r) return 0;
int mid = l + (r - l >> 1);
//归并到最底部也都是一左一右的情况
LL res = merge_sort(l, mid) + merge_sort(mid + 1, r);
//合并成一个数组
int k = 0, i = l, j = mid + 1;
while(i <= mid && j <= r)
{
if(q[i] <= q[j]) tem[k++] = q[i++];
else {
tem [k++] = q[j++];
res += mid - i + 1;
//此情况,左边数组该位置i以及后面的所有数都和j位置的数构成逆序对,因为此时两边都已经排好序了
}
}
//如果有剩余没有加入tem临时数组的,全部加上
while(i <= mid) tem[k++] = q[i++];
while(j <= r) tem[k++] = q[j++];
//将临时数组tem赋值给q
//两边的局部有序才能保证两边的整体有序
for(i = l, j = 0; i <= r; i++, j++) q[i] = tem[j];
return res;
}
int main()
{
cin >> n;
for(int i = 0; i< n; i++) cin >> q[i];
cout << merge_sort(0, n-1) << endl;
return 0;
}
基础算法:归并排序——逆序对的数量
于 2023-08-11 17:02:16 首次发布