蓝桥杯PREV-31 小朋友排队 【树状数组】

时间限制：1.0s   内存限制：256.0M

问题描述

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

输入格式

　　输入的第一行包含一个整数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。

解题思路

这题需要用到树状数组最基础的东西，不懂树状数组的先看这：https://blog.csdn.net/int64ago/article/details/7429868。

思路：

树状数组用到的两个数组在这里分别称为数据数组和统计数组，我们依次让小朋友入队，当第i个小朋友入队时，如果i前边有k个比i高的，那么i最少需要交换k次。那么如何求有多少个比i高的小朋友呢？ 我们这里以c[ j]表示身高 <= j 的人数。当第i个小朋友入队时，前边比i高的小朋友的个数即为 i - c[ h[i]] 。 他不仅需要和前边比他高的人换，还需要和后边比他矮的人换，那么如何计算向后换的最少次数呢？ 这里只需要把入队的顺序倒过来，即原来第一个入队的小朋友最后一个入队，原来最后一个入队的小朋友第一个入队。为什么要这么做呢？ 你想，我们先让原来最后一个入队的小朋友第一个入队，当倒数第i个小朋友入队的时候，c[ h[i]-1] 就是队里比i矮的人的人数了，也即正序入队时这个小朋友后边比他高的人数，即他需要向后交换的最少次数。 

AC代码

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+5;
long long h[maxn],c[maxn],num[maxn],tot[maxn];
// a小朋友身高	c统计数组	num记录第i个小朋友的交换次数	tot 记录交换第i次时的不高兴程度 
void init()
{
	tot[0]=0;
	for(long long i=1;i<maxn;i++) 
		tot[i] = tot[i-1]+i;	
}
int lowbit(int x)
{
	return x&(-x);
}
void Updata(int x)
{
	while(x < maxn) {
		c[x] += 1;
		x += lowbit(x);
	}
}
long long Get_sum(int x)
{
	long long sum=0;
	while(x > 0) {
		sum += c[x];
		x -= lowbit(x);
	}
	return sum;
}
int main()
{
	int n;
	cin >> n;
	memset(num,0,sizeof num);
	memset(c,0,sizeof c);
	init();
	for(int i=1;i<=n;i++) {
		cin >> h[i];	//读入第i个小朋友的身高 
		Updata(h[i]+1);		//树状数组下标从1开始，避免身高为0的情况。 
		num[i] += i - Get_sum(h[i]+1);	// i前边有多少个比i高的 
	}
	memset(c,0,sizeof c);
	for(int i=n;i>0;i--) {
		Updata(h[i]+1);
		num[i] += Get_sum(h[i]);	//i后边有多少个比i矮的 
	}
	long long sum=0;
	for(int i=1;i<=n;i++)
		sum += tot[num[i]];
	cout << sum;
}