hdu1394(线段树求逆序数)

hdu1394(线段树求逆序数)

题意

给出一串数字a，可以进行如下操作：将该串数字第一个数字a0放到该串数字的最后位置，形成新的数字串，求操作过程中最小逆序数。

思路

1）求一串数字的逆序数：
暴力法：对每一个数字x，判断x与x之前出现了多少个比x大的数字。复杂度为O(nn).
利用线段树：因为数字串出现的数字只能在区间0~n，可以利用线段树对出现的数字进行统计。对每个x，查询区间 [x+1,n-1] 已经有多少个被计数的数字，它们的合计就是逆序数。
2）用初始的逆序数推到移动第一个数到最后之后的逆序数
设初始逆序数为sum,
若去掉第一个数m，那么逆序数减少：m，
把m放到后面，逆序数增加：n - 1 - m,
所以：sum = sum - m + n - 1 - m = sum - 2m + n - 1。

AC代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 5e3+3;
int tree[maxn<<2];//最底层的叶子节点对应0~n-1相应数字，每次输入一个数字，就要计数
#define lson l, mid, rt<<1
#define rson mid+1, r, rt<<1|1
void build(int l, int r, int rt)
{
    tree[rt] = 0;
    if(l == r)return;
    int mid = (l+r) >> 1;
    build(lson);
    build(rson);
}

void update(int val, int l, int r, int rt)//val对应的叶节点以及区间加一。
{
    tree[rt]++;
    if(l == r)return;
    int mid = (l+r) >> 1;
    if(val <= mid)
        update(val, lson);
    else
        update(val, rson);
}

int query(int L, int R, int l, int r, int rt)//查询对应区间的计数
{
    if(L > r)return 0;
    if(L <= l && r <= R)return tree[rt];

    int mid = (l+r) >> 1;
    int ret = 0;
    if(L <= mid)ret += query(L, R, lson);
    if(R > mid)ret += query(L, R, rson);
    return ret;
}

int main()
{
    //freopen("in.txt", "r", stdin);
    int n;
    int a[maxn];
    while(~scanf("%d", &n)){
        build(0, n-1, 1);//建立区间树，初始化
        int sum = 0;
        for(int i = 1; i <= n; i++){
            scanf("%d", &a[i]);
            update(a[i], 0, n-1, 1);
            sum += query(a[i]+1, n-1, 0, n-1, 1);
        }
        int ans = sum;
        for(int i = 1; i < n; i++){
            sum = sum - 2*a[i] + n - 1;
            ans = min(sum, ans);
        }
        printf("%d\n", ans);
    }

    return 0;
}