洛谷P1908 && 蓝桥杯算法提高 逆序对
利用归并排序思路解决逆序对问题
(二分思想)
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#if 0
Goes && G.S.M.
Title://逆序对数量,归并排序 O(nlogn)
//思路:对于相邻两段,有序的a[l,mid]与有序的a[mid+1,r]
//可以O(r-l+1)计算逆序对数量
#endif
/*i,j分别为两段内的下标,如果有A[i] > A[j],
那么左边A[i]之后的元素也都会大于A[j],
所以A[j] 与 A[i] 到A[mid]的逆序对为 mid-i+1;*/
#define ll long long
const int N=5e5+5;
int a[N],n,temp[N];
ll query(int l,int r){
int mid=(l+r)/2;ll res=0;
if(l==r) return 0;
res=query(l,mid)+query(mid+1,r);
int i=l,j=mid+1,k=l-1;
while(i<=mid&&j<=r){
if(a[i]>a[j]){
res+=mid-i+1;
temp[++k]=a[j];
j++;
}
else{
temp[++k]=a[i];
i++;
}
}k++;
if(j<=r) for(k;k<=r;k++) temp[k]=a[j++];
else for(k;k<=r;k++) temp[k]=a[i++];
k--;
for(int i=l;i<=k;i++)
a[i]=temp[i];
return res;
}
int main(){
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
cout<<query(1,n);
// cout<<endl;for(int i=1;i<=n;i++) cout<<a[i]<<" ";
}