**788 acwing 逆序对的数量
给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000,
数列中的元素的取值范围 [1,10^9]。
输入
6
2 3 4 5 6 1
输出
5
想法:
第一次做这道题时,想采用两层for进行暴力计算,结果运行结果超时,最坏的时间复杂度为O(n^2),看了y总的题解后恍然大悟
思路:
为了减少时间复杂度,可采用二路归并,将数组分成 L ~ mid , mid+1 ~ r
设左边的数为q[i] L <= i <= mid, 右边的数为q[j] , mid+1 <= j <= r;只要计算去q[i]大于q[j]有多少个,即可计算出逆序对的数量,且左右两侧排完序后不影响上一层数组的逆序对的计算,该算法相当于二叉树的深度优先遍历;
int res = 0 ; // 设置初始逆序对为0
LL merge_sort(int* arr, int l, int r){
if (l >= r) return 0;
int mid = l + r >> 1, i = l, j = mid + 1; //
res = merge_sort(arr, i, mid) + merge_sort(arr, j, r); //计算左子树的逆序对数和右子树的逆序对数
int k = 0;
while (i <= mid && j <= r) {
if (arr[i] <= arr[j]) temp[k++] = arr[i++];
else {
temp[k++] = arr[j++];
res += mid - i + 1; //只要发现q[i] > q[j] 只要在 i ~ mid 范围中的数都大于q[j], 因此该节点对应的逆序对的数量为 res+mid-i+1
}
}
while (i <= mid)temp[k++] = arr[i++]; // 扫尾
while (j <= r) temp[k++] = arr[j++]; // 扫尾
for (i = l, k = 0; i <= r; i++, k++) {
arr[i] = temp[k]; // 顺序存储
}
return res; // 返回上一层
}