Description
n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。
每个小朋友都有一个不高兴的程度。开始的时候,所有小朋友的不高兴程度都是0。
如果某个小朋友第一次被要求交换,则他的不高兴程度增加1,如果第二次要求他交换,则他的不高兴程度增加2(即不高兴程度为3),依次类推。当要求某个小朋友第k次交换时,他的不高兴程度增加k。
请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。
Input
输入的第一行包含一个整数n,表示小朋友的个数。
第二行包含 n 个整数 H1 H2 … Hn,分别表示每个小朋友的身高。
Output
输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。
Sample Input
3
3 2 1
Sample Output
9
HINT
对于100%的数据,1<=n<=100000,0<=Hi<=1000000。
看见题目描述相信第一想法是暴力求解必然会超时
而又能发现小朋友最终从低到高排列,而交换的是不满足前小后大这一规律很容易想到逆序数。小朋友最小交换次数就是该小朋友在所有逆序对中出现的次数和,而交换k次不高兴程度就增加k,所以最后每个小朋友的不高兴程度就是1到其交换次数k的和。这样只要能统计处该小盆友在所有逆序对(比如3,2,1中2在<3,2>,<2,1>中都出现所以它的k就为2)中出现的次数就能解题。
所以简单地讲就是归并求下逆序数,然后换位置的时候两个人的不开心程度都要增加。最后求个和就好了。。
#include<bits/stdc++.h>
using namespace std;
#define mem(a,b) memset(a,b,sizeof a);
typedef long long ll;
const int maxn=100010;
struct node
{
int height;
long long cou;
} a[maxn],b[maxn];
void Merge(int start,int mid,int en)
{
int i=start,j=mid+1,k=start;
while(i<=mid&&j<=en)
{
if(a[i].height>a[j].height)
{
a[j].cou+=mid-i+1;
b[k++]=a[j++];
}
else
{
a[i].cou+=j-mid-1;
b[k++]=a[i++];
}
}
while(i<=mid)
{
a[i].cou+=en-mid;
b[k++]=a[i++];
}
while(j<=en)
b[k++]=a[j++];
for(i=start; i<=en; i++)
a[i]=b[i];
}
void mergesort(int start,int en)
{
if(start<en)
{
int mid=(start+en)/2;
mergesort(start,mid);
mergesort(mid+1,en);
Merge(start,mid,en);
}
}
int main(void)
{
int n;
scanf("%d",&n);
long long sum=0,temp;
for(int i=0; i<n; i++)
{
scanf("%d",&a[i].height);
a[i].cou=0;
}
mergesort(0,n-1);
for(int i=0; i<n; i++)
{
if(a[i].cou>0)
{
temp=(a[i].cou+1)*a[i].cou;
sum+=temp/2;
}
}
printf("%lld\n",sum);
return 0;
}