蓝桥 历届试题 小朋友排队

问题描述：传送门

　　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。

---

真坑，我还以为我树状数组不用复习了的。

首先说下我踩了的坑点：

1：看题第一反应逆序对，求次数，没想到（不会）还要反着求后面的顺序对（后面交换对前面的影响）。

2：我开始看每次+2，以为是公差为2的等差数列求和，我这阅读理解，真是，他是公差为1，只是要求和。

3：在getsum的时候，是会把本身给算进去的，在求逆序对的时候，用i-getsum(i),会把本身给减去，所以逆序对减去就是前面比i小的个数，但是在求顺序对的时候，直接+getsum(i)会把本身也给加上，所以是+getsum(i-1)。

4：说出来可能不信，这题身高还有0的，我就一直报超时，0&(-0)，这是什么啊，然后就一直死循环了，所以在要在所有原先输入的值上都+1（很多题都是这种情况，我也不是第一次见了，可是写的时候就是没考虑到）。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1000005;
int a[maxn],c[maxn];
long long ans[maxn];
int n;
int lowbit(int x){
	return x&(-x);
}
void updata(int a,int k){
	for(int i=a;i<maxn;i+=lowbit(i)){
		c[i]+=k;
	}
}
int getsum(int x){
	int ans=0;
	for(int i=x;i;i-=lowbit(i)){
		ans+=c[i];
	}
	return ans;
}
void init(){
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		updata(a[i]+1,1);
		ans[i]+=(i+1)-getsum(a[i]+1);
	}
}
void work(){
	memset(c,0,sizeof c);
	for(int i=n-1;i>=0;i--){
		updata(a[i]+1,1);
		ans[i]+=getsum(a[i]);//其实是ans[i]+=getsum(a[i]+1-1); 
	}

	long long sum=0;
	for(int i=0;i<n;i++){
		sum+=(1+ans[i])*ans[i]/2;
	}
	printf("%lld\n",sum);
}
int main(){
	init();
	work();
	return 0;
}