[个人记录] AcWing 788 逆序对的数量

788. 逆序对的数量

---

VIEDIO

#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
using namespace std;
const int N = 1e5+10;
typedef long long LL;
/*   BOKI   */
int tmp[N];
LL merge_sort(int a[],int l,int r) {
	 if(l >= r) return 0;
	 int mid = l+r >> 1;
	 //正常的归并排序
	 LL res = merge_sort(a,l,mid) + merge_sort(a,mid+1,r);
	 int k = 0,i = l,j = mid+1;
	 while(i <= mid && j <= r) {
	 	if(a[i] <= a[j]) tmp[k++] = a[i++];
	 	else {
	 		tmp[k++] = a[j++];
	 		res += mid-i+1;
	 		/*此时数组中两边都已经排序好，
	 		若存在 a[i] > a[j]说明 a[j] 小于所有 i ~ mid 中的数，
	 		因此此时res += mid -i + 1  
	 		下一次循环 j 已经 +1 了，因此之后的循环不会跑到上一个a[i] > a[j] 的 a[j]*/
	 	}
	 }
	 //防止没排完剩下
	 while(i <= mid) tmp[k++] = a[i++];
	 while(j <= r) tmp[k++] =  a[j++];
	 //把排好序的数组存回a[]，这就是为啥之前说数组两边都已经排好序了
	 for(int i = l,j = 0;i <= r;i++,j++) a[i] = tmp[j];
	 return res;
}

int main() {
	int n,a[N];cin >> n;
	for(int i = 0; i < n; i ++) cin >> a[i];
	cout << merge_sort(a,0,n-1) << endl;
	return 0;
}