Minimum Inversion Number HDU - 1394 线段树

题解

题目大意 给n个数字 这n个数字是一个环 问旋转到哪个位置逆序对的数量最少

使用线段树求出数列的逆序对数量 每次"旋转"减少开头数字造成的贡献 也就是a[i] - 1个 因为他能产生逆序的都是在他后面并且比他小的 再加上他添加到最后的贡献即可

AC代码

#include <stdio.h>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int MAXN = 5e3 + 10;
int a[MAXN];

struct node
{
	int l, r;
	int v;
}tre[MAXN * 4];

inline void PushUp(int x)
{
	tre[x].v = tre[x << 1].v + tre[x << 1 | 1].v;
}
void Build(int x, int l, int r)
{
	tre[x].l = l, tre[x].r = r, tre[x].v = 0;
	if (l == r)
		return; //=a[i]
	else
	{
		int m = l + r >> 1;
		Build(x << 1, l, m);
		Build(x << 1 | 1, m + 1, r);
		//PushUp(x);
	}
}
void Update(int x, int p, int v) //long long
{
	int l = tre[x].l, r = tre[x].r;
	if (l == r)
		tre[x].v += v;
	else
	{
		int m = l + r >> 1;
		if (m >= p)
			Update(x << 1, p, v);
		else
			Update(x << 1 | 1, p, v);
		PushUp(x);
	}
}
int Query(int x, int pl, int pr) //long long
{
	int l = tre[x].l, r = tre[x].r;
	if (pl <= l && r <= pr)
		return tre[x].v;
	else
	{
		int m = l + r >> 1;
		int v = 0; //long long
		if (m >= pl)
			v += Query(x << 1, pl, pr);
		if (m < pr)
			v += Query(x << 1 | 1, pl, pr);
		return v;
	}
}
int main()
{
#ifdef LOCAL
	freopen("C:/input.txt", "r", stdin);
#endif
	int n;
	while (cin >> n)
	{
		for (int i = 1; i <= n; i++)
			scanf("%d", &a[i]), a[i]++; //调整为1~n
		Build(1, 1, n);
		int res = 0, ans;
		for (int i = 1; i <= n; i++) //计算逆序对
		{
			res += Query(1, a[i], n);
			Update(1, a[i], 1);
		}
		ans = res;
		for (int i = 1; i < n; i++) //尝试旋转
		{
			res -= a[i] - 1; //删去第一个数字减少的逆序
			res += Query(1, a[i], n) - 1; //添加到最后增加的逆序
			ans = min(ans, res);
		}
		cout << ans << endl;
	}

	return 0;
}