今天温习树状数组,果然忘记了好多,树状数组求逆序数,值得注意这道题所有的数都是0-n-1的,所以在求最小的时候不用每个数顺序在计算一遍,我已开始就是把每个顺序又计算了一遍,果断超时了。第i个数拿到后面去,逆序数会减少a[i]-1,同时会增加n-a[i]
#include<iostream>
#include<stdio.h>
using namespace std;
int a[5005],tree[5005],n;
int lowbit(int x){
return x&(-x);
}
void add(int i,int val){
while(i<=n){
tree[i]+=val;
i+=lowbit(i);
}
}
int sum(int i){
int r=0;
while(i>=1){
r+=tree[i];
i-=lowbit(i);
}
return r;
}
int main(){
// freopen("in.txt","r",stdin);
while(~scanf("%d",&n)){
for(int j=0;j<=n;j++)
tree[j]=0;
int ans=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
a[i]++;
ans+=sum(n)-sum(a[i]);
add(a[i],1);
}
int min=ans;
for(int i=1;i<=n;i++){
ans=ans-(a[i]-1)+n-a[i];//
// cout<<ans<<endl;
if(ans<min) min=ans;
}
cout<<min<<endl;
}
}