2014年蓝桥杯预赛第十题小朋友排队

标题：小朋友排队

    n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。

    每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是0。

    如果某个小朋友第一次被要求交换，则他的不高兴程度增加1，如果第二次要求他交换，则他的不高兴程度增加2（即不高兴程度为3），依次类推。当要求某个小朋友第k次交换时，他的不高兴程度增加k。

    请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。

    如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。

 

【数据格式】

    输入的第一行包含一个整数n，表示小朋友的个数。

    第二行包含 n 个整数 H1 H2 … Hn，分别表示每个小朋友的身高。

    输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。

例如，输入：

3

3 2 1

程序应该输出：

9

 

解题思路：由于只能相邻元素之间交换，又要记录每个元素的交换次数来计算不高兴程度，涉及逆序数：在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。

只要记录每个元素在该排列中前面有几个大于它的数a(要往前移动几位)，排列中后面有几个小于它的数b(要往后移动几步)，a+b即为移动的次数，考虑到数据量巨大，可通过数组数组或线段树来解决。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int MAXN = 100005;
int arr[MAXN], tot[MAXN], n;	//swapcnt[]用于记录每个数要移动的总共次数 
LL tree[MAXN], swapcnt[MAXN];
int lowbit(int x){
	return x&(-x);
}

void UpdateTree(int x, int c){
	for(int i=x; i<=n; i+=lowbit(i)){
		tree[i] += c;
	}
}

LL getSum(int x){
	LL res = 0;
	for(int i=x; i>0; i-=lowbit(i)){
		res += tree[i];
	}
	return res;
}

void discrete(){
	sort(tot+1, tot+1+n);
	
	int m = unique(tot+1, tot+1+n) - (tot+1);
	
	for(int i=1; i<=n; ++i){
		arr[i] = lower_bound(tot+1, tot+1+n, arr[i]) - (tot+1) +1;
	}
}

int main()
{
	scanf("%d", &n);
	for(int i=1; i<=n; ++i){
		scanf("%d", &arr[i]);
		tot[i] = arr[i];
	}
	
	discrete();		//离散化 
	
	//获得在该序列中每个数前面有几个数比他大？
	for(int i=1; i<=n; ++i){
		UpdateTree(arr[i], 1);
		swapcnt[i] = i-getSum(arr[i]);		//getSum(arr[i])所得为[1,arr[i]]之间小于等于arr[i]数的个数, i为以arr[i]结尾的子序列长度 
	} 
	memset(tree, 0, sizeof(tree));
	//获得在该序列中每个数后面有几个数比他小？
	for(int i=n; i>=1; --i){
		UpdateTree(arr[i], 1);
		swapcnt[i] += getSum(arr[i]-1);		//因为这里是逆序加进来的，getSum(arr[i]-1)代表[arr[i],n]之间小于他的数的个数 
	} 
	
	LL ans = 0;
	for(int i=1; i<=n; ++i){
		ans += (1+swapcnt[i])*swapcnt[i]/2;
	} 
	
	printf("%lld", ans);
	return 0;
 }