每个小朋友都有一个不高兴的程度。开始的时候,所有小朋友的不高兴程度都是0。
如果某个小朋友第一次被要求交换,则他的不高兴程度增加1,如果第二次要求他交换,则他的不高兴程度增加2(即不高兴程度为3),依次类推。当要求某个小朋友第k次交换时,他的不高兴程度增加k。
请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。
第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。
3 2 1
对于30%的数据, 1<=n<=1000;
对于50%的数据, 1<=n<=10000;
对于100%的数据,1<=n<=100000,0<=Hi<=1000000。
问题的关键在于查找逆序对的个数,最容易想到的当然就是暴力大法,看看数据范围,超时无疑,于是我们就需要使用一个特殊的数组----树状数组。
首先我们看看树状数组的特点,当然就是很快速的求前n个元素的和,那这和我们求逆序对有什么关系呢?
这里,我们要巧妙地变一下形,还是以题目中给出的数据为例 3 2 1
因为树状数组实际上是由两部分组成:数据数组+统计数组,我们只看数据数组
由于树状数组是从1开始的,而题目中小盆友的身高可以为0(真是长见识了),所以我们将每个小盆友的身高加1然后作为树状数组的下标,将数值1存到相应的位置.
第一次读入3,此时读入的数据量为1个,变成这样
C[1] C[2] C[3] C[4] C[5] C[6] C[7] C[8]
0 0 0 1 0 0 0 0
可以看到sum(C[1],C[4])=1(可以由树状数组的统计数组得到),这个是小于等于3的数字的个数,也就是说当输入第一个数字3的时候没有比它小的数字存在,这时我们用 输入数字总数-sum(C[1],C[4])=0,也就是说大于3的数字的个数为0,我们令b[0]=0.
第二次读入2,此时读入的数据量为2个,变成这样
C[1] C[2] C[3] C[4] C[5] C[6] C[7] C[8]
0 0 1 1 0 0 0 0
可以看到sum(C[1],C[3])=1,任然不存在比它小的数,但此时输入的数据总量为2,而2-1=1,就是说,存在一个数在2之前并且大于2,这个数当然就是3,我们另b[1]=1.
第三次读入1,此时读入的数据量为3,变成这样
C[1] C[2] C[3] C[4] C[5] C[6] C[7] C[8]
0 1 1 1 0 0 0 0
可以看到sum(C[1],C[2])=1,任然不存在比它小的数,但此时输入的数据总量为3,而3-1=2,就是说,存在两个数在1之前并且大于1,这个数当然就是2,3,我们另b[2]=2.
到此,我们已经算出了每个数前面的较大的数的个数了,数据存在num[]中,现在我们再反过来,先插入1,再插入2,再插入3,但这次我们不再用总数减去sum了,而直接求sum,求出来的自然就是,每个数后面的较小的数的个数,然后将得到的数值累加到相应的b[i]中,最终我们会得到b[0]=2,b[1]=2,b[2]=2,分别对应num[0]=3,num[1]=2,num[2]=1。
求得了每个小盆友被移动的次数,我们需要计算其不高兴程度,这里实际上可以事先打个表,就是将被移动n次后的不高兴值全算出来,然后直接用就可以了,这里,我们将其存到total[]数组中,而且total[2]=3,所以总不高兴值就是9.
需要注意的是如果重复的数字出现怎么办,如果出现,实际上出问题的会是求每个数之前较大数的那部分,因为用到了数的总个数,如果出现一样的数,就会导致相减后的结果偏大,而且正好是大了 重复量-1 ,那么我们就可以算出重复量,然后将这一部分减去就行,关键是怎么算重复量,实际也很简单,通过树状数组,我们可以求得sum(1,a)和sum(1,a+1),其中输入的数字为a,前者算出的小于a的数的个数,后者算出的是小于等于a的数的个数,两个一减就是等于a的个数。
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define SIZE 1000000+10
#define N 100000+10
int height[SIZE];//记录每个身高
int C[SIZE];//这是树状数组的统计数组
long long b[N];//记录每个人的前逆序对+后逆序对
int n;
long long judge[N];//不高兴度表judge[i]代表移动N次的不高兴度
int lowbit(int x){
return x&(-x);
}
void add(int pos,int num) {
while(pos<SIZE) {
C[pos]+=num;
pos+=lowbit(pos);
}
}
int sum(int end) {
int re=0;
while(end>0) {
re+=C[end];
end-=lowbit(end);
}
return re;
}
void fun() {
int i;
for(i=0; i<n; i++) {
cin>>height[i];
add(height[i]+1,1);
b[i]=(i+1)-sum(height[i]+1);
}
memset(C,0,sizeof(C));
for(int i=n-1; i>=0; i--) {
add(height[i]+1,1);
b[i]+=sum(height[i]);
}
long long ans=0;
for(int i=0;i<n;i++){
ans+=judge[b[i]];
}
cout<<ans;
}
int main() {
int i;
cin>>n;
memset(height,0,sizeof(height));//初始化
memset(C,0,sizeof(C));
memset(b,0,sizeof(b));
judge[0]=0;
for(i=1; i<N; i++) {
judge[i]=judge[i-1]+i;
}
fun();
return 0;
}