注意:给定的数据是0到n-1的一个排列。如果是任意数据则程序出错。
线段树节点只记录当前有多少个数。
当要查询a[i]时,只需要在当前线段树查询 a[i] – n-1之间有多少个数(都比a[i]大,逆序)就知道与前面与a[i]构成逆序数的对数。查询完成然后把a[i]插入到线段树,维护节点数。
/*hdu 1394 Minimum Inversion Number(线段树单点更新)
题目大意:
给定一组数,不停的将数列的第一个数放在数列的最后,
形成一个新的数列,找出所有数列中逆序数最小是多少
注意:此组数据是从0-n-1!不是随机数据!
思路:
只需要求出原始数列的逆序数即可,转化后的数列的逆序数
可以通过原始数列的逆序数得出。
逆序数求法:
将数列的每个点依次加入到线段树中,每个叶子节点sum[]==1
表示该点已加入到线段树中,非叶子节点表示该区间中已加入线段树
的数的个数。
查询时,假设查询的数为i,那么就查找区间[i,n](n是点的个数)
中已加入线段树中的数的总数,即该点的逆序数。查询i时线段树中的数
都是数列中在i前面的数,那么只需查找比i大的区间的数的个数
*/
#include <cstdio>
#include <algorithm>
#include <string.h>
using namespace std;
#define lson l, m, root<<1
#define rson m+1, r, root<<1|1
const int N = 5100;
int sum[N<<2];
int a[N];
void build(int l, int r, int root)
{
sum[root] = 0;
if(l == r) return;
int m = (l + r) >> 1;
build(lson);
build(rson);
}
void update(int pos, int l, int r, int root)
{
if(l == pos && pos == r)
{
sum[root] ++;
return;
}
int m = (l + r) >> 1;
if(pos <= m) update(pos, lson);
if(pos > m) update(pos, rson);
sum[root] = sum[root<<1] + sum[root<<1|1];
}
int query(int L, int R, int l, int r, int root)
{
if(L <= l && r <= R)
return sum[root];
int m = (l + r) >> 1;
int ans = 0;
if(L <= m)
ans += query(L, R, lson);
if(m < R)
ans += query(L, R, rson);
return ans;
}
int main()
{
int n,i;
freopen("a.txt","r",stdin);
while(~scanf("%d",&n))
{
build(0, n-1, 1);
int cnt = 0;
for(i = 0; i < n; i ++)
{
scanf("%d",&a[i]);
cnt += query(a[i], n-1, 0, n-1, 1);
update(a[i], 0, n-1, 1);
}
int ans = cnt;
for(i = 0; i < n; i ++)
{
cnt = cnt + (n - a[i] - 1) - a[i];
//上一个序列的逆序对数加上比a[i]大的(a[i]放到最后一个位置后,
//能与前面每一个比它大的构成逆序对)再减去比a[i]小的
ans = min(ans, cnt);
}
printf("%d\n",ans);
}
return 0;
}