题意:给你一个从 0~n-1 的排列,通过一次或多次把第一个数移动到最后一个数的位置(为了方便,后文中我们称这种变换为 P 变换),将得到许多的逆序数(最多n-1个)的逆序数,求所有得到的逆序数中的最小的一个。
解题思路:如果把每个排列的逆序数暴力求出来,会超时,时间复杂度将达到o( n^3 ),肯定超时。 所以这里我们可以发现一个规律,设第一个数为 a[ i ],通过一次P变换,得到的逆序数和上次比,多了 n - 2 * a[ i ] -1(这个数可能为负数)。
为什么呢? 在整个排列中, 比 a[ i ] 小的数从 0 ~ a[ i ] - 1,共 a[ i ]个,比a[ i ] 大的数有 n -1 - a[ i ]个,经P变换后,之前比a[ i ] 大的数会使得逆序数共增加 n - 1 - a[i] ,之前比a[ i ]小的数会使得变换后的逆序数共减小 a[ i ],所以, 一共变化了 n-1-a[ i ]-a[ i ] , 即 n - 2 * a[ i ] -1。
有了这个规律,接下来我们可以温柔暴力了,即暴力求出第一个排列的逆序数(这个还是比较简单的吧),根据这个规律,我们可以在o( n^2 )内求出这个答案。
不过如果使用线段树,会有更优答案(线段树降低的复杂度到 o( n*lgn ))。
代码如下:
#include<stdio.h>
#include<algorithm>
#define N 5005
#define mid(a,b) ((a+b)>>1)
using namespace std;
struct{
int l, r;
int v;
}segTree[N*4];
void build(int root, int left, int right){
segTree[root].l = left;
segTree[root].r = right;
segTree[root].v = 0;
if(left == right)
return;
build(root*2, left, mid(left,right));
build(root*2+1, mid(left,right) + 1, right);
}
void minsert(int root, int pos){
segTree[root].v ++;//
if(segTree[root].l == pos && segTree[root].r == pos)
return;
if(pos <= mid(segTree[root].l,segTree[root].r))
minsert(root*2, pos);
else
minsert(root*2+1, pos);
}
int query(int root, int left, int right){
if(segTree[root].l == left && segTree[root].r == right)
return segTree[root].v;
if(right <= mid(segTree[root].l, segTree[root].r))
return query(root*2, left, right);
else if( left > mid(segTree[root].l, segTree[root].r))
return query(root*2+1, left, right);
else
return query(root*2, left, mid(segTree[root].l, segTree[root].r)) + query(root*2+1, mid(segTree[root].l, segTree[root].r) + 1, right);
}
int main(){
int n, a[N];
while(~scanf("%d", &n)){
build(1,0,n-1);
int sum = 0;
for(int i = 0; i < n; i++){
scanf("%d", &a[i]);
sum += query( 1, a[i], n-1 );
minsert(1, a[i]);
}
int ans = 10000000;
for(int i = 0; i < n; i++){
sum = sum + n - 2*a[i] - 1;
if(sum < ans)
ans = sum;
}
printf("%d\n", ans);
}
return 0;
}
我们在节点结构体中多一个元素 v,来保存该节点目前存有多少数(当数插入(minsert)时,V++)。
查询是从a[ i ] ~ n - 1的,即查询比a[ i ]大的数目前已经存进去多少个。 查询完以后,我们将a[ i ]插入线段树中。
注意,在程序中,我们每次都是查询完以后再进行插入操作的。
这样最后就得到了第一个排列的逆序数(n*lgn)。
再通过之前的规律,即可得出答案。