历届试题 小朋友排队 逆序对（树状数组）

  题目链接                  蓝桥杯---历届真题 题解

问题描述

　　n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。
　　每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是0。
　　如果某个小朋友第一次被要求交换，则他的不高兴程度增加1，如果第二次要求他交换，则他的不高兴程度增加2（即不高兴程度为3），依次类推。当要求某个小朋友第k次交换时，他的不高兴程度增加k。
　　请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。
　　如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。

 分析：在n个序列中的a，左边有x个比a高的，右边有y个比a矮的则，a的交换次数为x+y次（题意为1到x+y的和 = （x+y)*(x+y+1)/2）。这就需要求逆序对。可以用树状数组求逆序对，树状数组能动态维护和查询，时间复杂度O(NlogN）。好像归并排序也能做，可以试一试。

方法一：树状数组，O(NlogN）

#include <cstdio>
#include <cstring>
const int N = 100000+5, H = 1000000+5;
int num[N], c[H],sum[H]; 
int lowBit(int x){
	return x&-x;
}

int getSum(int x){
	int res = 0;
	for(int i = x; i; i -= lowBit(i))
		res += c[i];
	return res;
}
//更新操作 
void add(int x,int d){
	for(int i = x; i < H; i += lowBit(i))
		c[i] += d; 
}

int main(int argc, char** argv) {
	int n;
	scanf("%d",&n);
	for(int i = 0; i < n; i++){ //左边逆序对 
		scanf("%d",&num[i]);
		add(num[i]+1, 1); //注意树状数组是从1开始的； 
		sum[i] += i+1 - getSum(num[i]+1); //找到<=num[i]+1的个数，总的个数减去即为>num[i]+1的个数 
	}
	memset(c,0,sizeof(c));
	for(int i = n-1; i >= 0; i--){ //反过来就是右边逆序对 
		add(num[i]+1,1);
		sum[i] += getSum(num[i]);
	}
	long long ans = 0;
	for(int i = 0; i < n; i++)
		ans += (long long)(1 + sum[i]) * sum[i] / 2; //10^6*10^6在int中会溢出 
	printf("%lld\n",ans);
	return 0;
}

输入格式

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

输出格式

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

样例输入

3
3 2 1

样例输出

9

样例说明

　　首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。

数据规模和约定

　　对于10%的数据， 1<=n<=10；
　　对于30%的数据， 1<=n<=1000；
　　对于50%的数据， 1<=n<=10000；
　　对于100%的数据，1<=n<=100000，0<=Hi<=1000000。