#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 5005
int a[N];
int sum[N<<2];
void build(int l,int r,int rt){
if(l==r){
sum[rt]=0;
return;
}
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
sum[rt]=0;
}
void update(int l,int r,int x,int rt){
if(l==r){
sum[rt]=1;
return;
}
int mid=(l+r)>>1;
if(x<=mid) update(l,mid,x,rt<<1);
else update(mid+1,r,x,rt<<1|1);
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int query(int l,int r,int st,int ed,int rt){
if(l>=st&&r<=ed){
return sum[rt];
}
int mid=(l+r)>>1;
int temp=0;
if(st<=mid) temp+=query(l,mid,st,ed,rt<<1);
if(ed>mid) temp+=query(mid+1,r,st,ed,rt<<1|1);
return temp;
}
int main(){
int n;
while(scanf("%d",&n)!=EOF){
build(0,n-1,1);
int cnt=0;
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
cnt+=query(0,n-1,a[i],n-1,1);
update(0,n-1,a[i],1);
}
//printf("%d\n",cnt);
int minn=cnt;
for(int i=0;i<n;i++){
cnt+=n-1-a[i]-a[i];
minn=min(minn,cnt);
}
printf("%d\n",minn);
}
return 0;
}
上面是线段树写法,顺便复习下归并排序如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N =5005;
const int inf=0x7f7f7f7f;
int a[N];
int b[N];
int temp1[N],temp2[N];
int num;
void merge(int l,int mid,int r){
for(int i=l;i<=mid;i++){
temp1[i]=a[i];
}
temp1[mid+1]=inf;
for(int i=mid+1;i<=r;i++){
temp2[i]=a[i];
}
temp2[r+1]=inf;
int ans=l;
int i=l;
int j=mid+1;
while(ans<=r){
if(temp1[i]<=temp2[j]) a[ans++]=temp1[i++];
else {
num+=mid-i+1;
a[ans++]=temp2[j++];
}
}
}
void mergesort(int l,int r){
if(l>=r) return;
int mid=(l+r)>>1;
mergesort(l,mid);
mergesort(mid+1,r);
merge(l,mid,r);
}
int main(){
int n;
while(scanf("%d",&n)!=EOF){
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
b[i]=a[i];
}
num=0;
mergesort(0,n-1);
/*printf("%d\n",num);
for(int i=0;i<n;i++){
printf(" %d ",a[i]);
}
*/
int minn=num;
for(int i=0;i<n;i++){
num+=-b[i]+(n-b[i]-1);
minn=min(minn,num);
}
printf("%d\n",minn);
}
return 0;
}