历届试题 小朋友排队

  历届试题 小朋友排队  

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

    

问题描述

　　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。

//该题使用树状数组来求逆序数。

//对于一个小朋友,他的交换次数取决于他前面有多少个比他高的,他的后面有多少个比他矮的.就可以得到该小朋友的交换次数,进而计算得到不高兴程度.

//因为不高兴是相对的，ab交换，ab都不高兴，所以需要前后求两次逆序数。比如3 2 1，若只求一次逆序数，则交换次数：3的等于0,2的等于1,1的等于2，不高兴程度3的等于0,2的等于1,1的等于1+2=3；总和为4，不符合题意。

//树状数组c[]的下标是从1开始的，只能求1~x的和，而该题小朋友的身高是从0开始的，所以要用 （身高+1）来对应数组c的下标，方便求和。

#include<stdio.h>
#include<string.h> 
const int M=1e6+10;   //注意这种定义方法代表10的6次方

const int N=1e5+10;
int H[N],b[N],c[M];       //H数组对应树状数组中的a数组，b记录了每个小朋友交换的次数，c对应了树状数组的c数组，以备求和
int lowbit(int x)             //lowbit函数求的是2^k,k是层数，从第0层开始向上，k的值是x（二进制表示）从右往左第一个1的位置。

{                                        //第x项的项数就是2^k项
return x&-x;
}
int sum(int i)               //求和1~x的和
{
int ans=0;
while(i>0)
{
ans+=c[i];
i-=lowbit(i);
}
return ans;
}
void add(int d,int x)        //原意为修改c[x]的值，这里指的是一个个将输入的身高插入到数组c中，

                                               //以便计算第x个小朋友之前的比他矮的人数
{
while(x<=M)
{
c[x]+=d;        //该题中d的值始终为1
x+=lowbit(x);//树状数组中数组a一个值改变，会引起c的多个值改变
}
return;
}
int main()
{
int n,i;
scanf("%d",&n);
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
for(i=1;i<=n;i++)
{
scanf("%d",&H[i]);
add(1,H[i]+1);             //注意身高+1来对应下标
b[i]=i-sum(H[i])-1;     //减去比H[i]（+1）矮的人数 ，及他自己，i为当前插入的人数
b[i]-=sum(H[i]+1)-sum(H[i])-1;//再减去和H[i]一样高的人数 ，因为只能求出前i项和，所以求单独一项i的值，需要
    }//前i项的和减去前i-1项的和，此题还需要减去自己，剩下比他高的人 
    memset(c,0,sizeof(c));//c数组清零，倒着插入，求其后比其矮的人
for(i=n;i>=1;i--)
{
add(1,H[i]+1);
b[i]+=sum(H[i]);//比H[i]+1矮的人数 
} 
long long r=0;//数比较大，用long long
for(i=1;i<=n;i++)
{
r+=(long long)b[i]*(b[i]+1)/2;//等差数列求和，需要类型强制转换，要不然存不下
}//每个小朋友i的交换次数为b[i],1~b[i]等差数列求和，再求n个小朋友的总和
printf("%lld\n",r);
return 0;
 }