求逆序对 - 归并排序
程序员文章站
2022-05-10 11:51:47
...
题目连接:http://ybt.ssoier.cn:8088/problem_show.php?pid=1311
n开到1e5的话,硬刚肯定是要超时的,所以可以用归并排序,并在排序的时候求出结果,其特点如下:
在每次合并两个子数组的时候,这两个子数组都已经是有序的啦
然后若a[i]>a[j]的话ans+=m-i+1
a[i]>a[j],则a[i-m]都>a[j]所以ans+=m-i+1
并且在比较两个数组元素的过程中,元素的相对位置不变,所以结果是正确的
代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int INF=0x3f3f3f3f;
const int N=100005;
int a[N],b[N],n;
ll ans=0;
void Merge(int s[],int t[],int b,int m,int e){
int i=b,j=m+1,k=0;
while(i<=m&&j<=e){
if(s[i]>s[j]){
ans+=m-i+1;
t[k++]=s[j++];
}
else t[k++]=s[i++];
}
while(i<=m)t[k++]=s[i++];
while(j<=e)t[k++]=s[j++];
for(int q=0;q<k;q++){
s[q+b]=t[q];
}
}
void MSort(int s[],int t[],int b,int e){
if(b<e){
int m=(b+e)>>1;
MSort(s,t,b,m);
MSort(s,t,m+1,e);
Merge(s,t,b,m,e);
}
}
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
MSort(a,b,0,n-1);
printf("%lld\n",ans);
}
上一篇: Android 零碎知识点---1
下一篇: 补充知识:断言