HDU 1394 Minimum Inversion Number（单点更新 + 求逆序数）

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1394

利用线段树求出当前序列的逆序数，然后O(1)的时间就可以求出每次操作后序列的逆序数。

代码如下。

#include <iostream>#include <iomanip>#include <cstdio>#include <algorithm>#include <cmath>#include <queue>#include <vector>#include <stack>#include <string>#include <cstring>#include <cassert>using namespace std;typedef long long ll;const int maxn=5005;const int INF=0x7fffffff;const int mod=1e7+7;#define LSON l,m,rt<<1#define RSON m+1,r,rt<<1|1#define ESP 1e-7int sum[maxn<<2],num[maxn];void pushup(int rt){    sum[rt]=sum[rt<<1]+sum[rt<<1|1];//每个节点存该区间已经输入的数字属于该区间的个数}void build(int l,int r,int rt){    sum[rt]=0;    if(l==r) return ;    int m=(l+r)>>1;    build(LSON);    build(RSON);}void update(int p,int l,int r,int rt){    if(l==r){        sum[rt]++;        return ;    }    int m=(l+r)>>1;    if(p<=m) update(p,LSON);    else update(p,RSON);    pushup(rt);}int query(int L,int R,int l,int r,int rt){    if(L<=l&&r<=R){        return sum[rt];    }    int m=(l+r)>>1,ans=0;    if(L<=m) ans+=query(L,R,LSON);    if(R>m) ans+=query(L,R,RSON);    return ans;}int main(){    int n;    while(~scanf("%d",&n)){        build(0,n-1,1);        int ss=0;//记录初始序列逆序数个数        for(int i=0;i<n;i++){            scanf("%d",&num[i]);            ss+=query(num[i],n-1,0,n-1,1);//对每次输入一个数后,查询已经输入的数字中比该数字的大的个数,即由该数字组成的逆序数个数            update(num[i],0,n-1,1);        }        int an=ss;        for(int i=0;i<n;i++){            ss+=n-num[i]-1-num[i];//当将第一个数放到最后时,总的逆序数个数=比它大的数-由它组成的逆序数个数.            an=min(an,ss);//记录最小的逆序数个数        }        printf("%d\n",an);    }    return 0;}