问题描述
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。
归并排序,计算逆序数,所有人分为上数组和下数组,这样每次都从两个数组的第一个开始比较,然后计算交换所需要的花费(逆序数),最后的结果让上数组接下数组组成一个正确序列,求出总逆序数就是小朋友的不开心值。
#include<iostream>
#include<cstdio>
using namespace std;
#define max 100005
struct point
{
int h,t;
};
point a[max],r[max],c[max];
int n;
void mery(point *a,int first,int last) //归并时,上数组的第一个元素与下数组的第一个元素相等,上数组的指针右移
{
int p,q,temp,mid;
mid=(first+last)/2;
temp=first;
p=first;
q=mid+1;
while(p<=mid&&q<=last)
{
if(a[p].h>a[q].h)
{
a[q].t+=mid+1-p;//上数组的第一个元素比下数组的第一个元素大,下数组的第一个元素前面有mid+1-p个数比他大
c[temp++]=a[q++];
}
else
{
a[p].t+=q-1-mid;
//上数组的第一个元素与下数组的第一个元素相等或者上数组的第一个元素比下数组的第一个元素小
//上数组的第一个元素与下数组的第一个元素相等时,也就是说下数组的第一个元素的前面的元素都比上数组的第一个元素小,上数组的第一个元素后面有q-1-mid个数比他小
c[temp++]=a[p++];
}
}
while(q<=last) //下数组有剩余
{
c[temp++]=a[q++];
}
q--;
while(p<=mid) //上数组有剩余,也就是说下数组的所有元素都比上数组剩下的元素小,故上数组每个元素t加q-mid,也就是下数组的元素个数
{
a[p].t+=q-mid;
c[temp++]=a[p++];
}
int i;
for(i=first; i<=last; i++)
{
a[i]=c[i];
}
}
void mmery(point *a,int first,int last)
{
if(first==last)
return;
int mid=(first+last)/2;
mmery(a,first,mid);
mmery(a,mid+1,last);
mery(a,first,last);
}
int main()
{
scanf("%d",&n);
for(int i=0; i<n; i++)
{
scanf("%d",&(a[i].h));
a[i].t=0;
}
mmery(a,0,n-1);
long long sum=0;
for(int i=0; i<n; i++)
{
long long s=a[i].t;
sum+=(s+1)*s/2;//求和
}
cout<<sum<<endl;
return 0;
}