首页 > 代码库 > HDU 1394 Minimum Inversion Number 树状数组

HDU 1394 Minimum Inversion Number 树状数组

今天温习树状数组,果然忘记了好多,树状数组求逆序数,值得注意这道题所有的数都是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;
    }
}