hdu 1394 Minimum Inversion Number （树状数组）

题目大意：给你一个1到n的排列（虽然题目是0到n-1的排列但加1一下就好了）

如是a1、a2、a3、.......、an-1、an

然后可以进行这样的操作：将第一个数放到最后，变成：a2、a3、a4、.......、an-1、an、a1

每次这样可以得到n个不同的串。

问这n个不同的串中，逆序对数最小的串的数量是多少。

 

这一题有一个关键的信息就是：这是一个1到n的排列

利用这个信息我们一可以O（1）的算出每一次操作后逆序对的变化数量：

如原串是a1、a2、......、an-1、an。逆序对数是：m

进行一次操作后，变成a2、a3、......、an-1、an、a1以后，原先（a1，*）的逆序对有n-a1个，新产生的（*，a1）的逆序对是a1-1个

 

所以这道题目的做法就是，先用树状数组求出一个串的逆序对。然后进行n次操作，并且在每次操作后用O（1）的时间复杂度算出新串的逆序对数。

 

我的代码：

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
using namespace std;

const long long MAXN = 1e6+1000;
int h[MAXN], a[MAXN], pos[MAXN];
int n;

int lowbit(int i)
{
	return i & (-i);
}
void updata(int i, int val)
{
	while (i <= n)
	{
		h[i] += val;
		i += lowbit(i);
	}
}
int query(int i)
{
	int ans = 0;
	while (i > 0)
	{
		ans += h[i];
		i -= lowbit(i);
	}
	return ans;
}

int main()
{
	long long ans, sum;
	int i;
	while (scanf("%d",&n)!=EOF)
	{
		for (i=1; i<=n; i++)
		{
			scanf("%d",&a[i]);
			a[i] ++;
			pos[a[i]] = i;
			h[i] = 0;
		}
		sum = 0;
		for (i=n; i>=1; i--)
		{
			updata(pos[i], 1);
			sum += 1ll * query(pos[i] - 1);
		}
		ans = MAXN*MAXN;
		for (i=1; i<=n; i++)
		{
			ans = min(ans, sum);
			sum -= a[i] - 1;
			sum += n - a[i];
		}
		printf("%I64d\n", ans);
	}
	return 0;
}