统计逆序对的两种解法
归并排序(mergeSort)
逆序对定义
\(i<j\) 但\(a[i]>a[j]\),假设我们分别使得通过mergeSort使得左右半边有序
即\(a[1]...a[mid]\) 递增, \(a[mid+1]....a[n]\)递增,我们需要通过merge操作,完成整个的排序和新增逆序对的计数,较小值出现在左半边记为 a[i],出现在右半边即为 a[j],那么每次出现在右半边,意味左半边比a[i]大的数都比a[j]大,由此可以统计逆序对
HDU1394
代码实现
#include<bits/stdc++.h>
using namespace std;
#define db(x) cout<<"["<<#x<<"]="<<x<<endl
/*
归并排序求逆序对+规律
*/
const int maxn = 5010;
int a[maxn];
int c[maxn];
int b[maxn];
//mergeSort
int n;
int merge1(int* a,int l1,int r1,int l2,int r2){
int p1=l1,p2 = l2;
int t = 0;
int cnt = 0;
//db(a[p1]);db(a[p2]);
while(p1<=r1&&p2<=r2){
if(a[p1]<a[p2]){
b[t] = a[p1];
p1++;
t++;
}
else{//a[p1]>a[p2]; a[p2] 小于 p1...r1所有数
b[t] = a[p2];
cnt+=(r1-p1+1);
//db(cnt);
p2++;
t++;
}
}
while(p1<=r1){b[t]=a[p1];p1++,t++;}
while(p2<=r2){b[t]=a[p2];p2++,t++;}
for(int k=0;k<t;k++){
a[l1+k] = b[k];
}
//db(cnt);db(l1);db(r1);db(l2);db(r2);
return cnt;
}
int mergeSort(int* a,int l,int r){
if(l==r) return 0;
int cnt = 0;
int mid = (l+r)>>1;
cnt+=mergeSort(a,l,mid);
cnt+=mergeSort(a,mid+1,r);
cnt+=merge1(a,l,mid,mid+1,r);
return cnt;
}
int main(){
while(cin>>n){
for(int i=0;i<n;i++){cin>>a[i];c[i]=a[i];}
int tmp = mergeSort(a,0,n-1);
//db(tmp);
int mint = tmp;
for(int i=0;i<n-1;i++){
tmp +=n-1-2*c[i];
//db(tmp);
mint = min(tmp,mint);
}
cout<<mint<<endl;
}
return 0;
}
线段树
线段树的解法非常简单,每次插入a[i] ,同时对a[i]+1....n-1进行计数;
此时要求元素范围不能太大,当然如果是在\(1..n\)之间,那么非常理想
代码实现
#include<bits/stdc++.h>
using namespace std;
#define db(x) cout<<"["<<#x<<"]="<<x<<endl
const int maxn = 5e3+10;
struct node{
int l,r,num; //num维护的信息是节点插入的区间插入节点的数目
}tr[maxn<<2];//线段树
int a[maxn];
void build(int n,int x,int y){//n是根节点下标,x,y是维护的区间范围
tr[n].l = x,tr[n].r = y;
tr[n].num = 0;
if(x==y) return ;
int mid = (x+y)>>1;//no over
build(n<<1,x,mid);
build(n<<1|1,mid+1,y);
tr[n].num = tr[n<<1].num+tr[n<<1|1].num;
}
void modify(int n,int p){//跟新区间单点p的信息
int l = tr[n].l, r= tr[n].r;
if(l==r&&l==p){//found
tr[n].num=1;
return ;
}
int mid = (l+r)>>1;
if(p<=mid) modify(n<<1,p);
if(p>mid) modify(n<<1|1,p);
tr[n].num = tr[n<<1].num+tr[n<<1|1].num;
}
int query(int n,int x,int y){
int l = tr[n].l , r=tr[n].r;
int mid = (l+r)>>1;
int ans = 0;
if(l>=x&&r<=y){//x,y覆盖了l,r
return tr[n].num;
}
if(x<=mid) ans+=query(n<<1,x,y);
if(y>mid) ans+=query(n<<1|1,x,y);
return ans;
}
int n;
int main(){
while(cin>>n){
build(1,0,n-1);
int ans = 0;
for(int i=0;i<n;i++){
cin>>a[i];
int t=query(1,a[i]+1,n-1);
//db(t);
ans+=t;
modify(1,a[i]);
}
int mint = ans;
//db(ans);
for(int i=0;i<n-1;i++){
ans+=(n-1-2*a[i]);
mint = min(ans,mint);
}
cout<<mint<<endl;
}
return 0;
}