HDU 1394 Minimum Inversion Number 树状数组

标签：os   io   for   re   c   amp   

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

HDU 1394 Minimum Inversion Number 树状数组,布布扣,bubuko.com

HDU 1394 Minimum Inversion Number 树状数组

标签：os   io   for   re   c   amp   

原文地址：http://blog.csdn.net/youngyangyang04/article/details/38069579