问题描述
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。
思路:模拟肯定不行,两重for循环在50%的数据的时候就完了,模拟效果只有30分。。然后看题解,有两种方法,一种是线段树和归并排序。每一个小朋友需要交换的次数是左边比之大的个数和右边比之小的个数。用到O(nlogn)的算法,而归并排序的思想就是划分为更小的区间,将之对比。然后需要注意的地方是,本题求的答案不是逆序数之和,而是每次交换都要加上上一次的结果加一。
代码:
#include<iostream>
#include<cstdio>
using namespace std;
struct point
{
int h;
int t;//统计该元素左边比他大的数的个数和该元素右边比他小的数的个数的总和
};
point l[110000],c[110000];
void Merge(point *a,int first,int last) //归并时,上数组的第一个元素与下数组的第一个元素相等,上数组的指针右移
{
int mid=(first+last)/2;
int temp=first;
int 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=first;
for(; i<=last; i++)
{
a[i]=c[i];
}
}
void MergeSort(point *a,int first,int last)
{
if(first==last)
{
return;
}
int mid=(first+last)/2;
MergeSort(a,first,mid);
MergeSort(a,mid+1,last);
Merge(a,first,last);
}
int main()
{
int n;
while(~scanf("%d",&n))
{
int i=0;
for(; i<n; i++)
{
scanf("%d",&(l[i].h));
l[i].t=0;
}
MergeSort(l,0,n-1);
long long sum=0;//数的范围
for(i=0; i<n; i++)
{
long long s=l[i].t;
sum+=(s+1)*s/2;//求和
}
cout<<sum<<endl;
}
return 0;
}
#include<iostream>
#include<cstdio>
using namespace std;
struct node
{
int a;
int b;
};
node child[110000],c[110000];
void Merge(node *a,int first,int last)
{
int mid = (first+last)/2;
int temp = first;
int p = first, q = mid+1;
//大小比较
while( p <= mid && q <= last)
{
if(a[p].a > a[q].a)
{
a[q].b += mid+1-p;
c[temp++]=a[q++];
}
else
{
a[p].b += q-1-mid;
c[temp++]=a[p++];
}
}
//数组剩余
while(q<=last)
{
c[temp++]=a[q++];
}
q--;
while(p<=mid)
{
a[p].b += q-mid;
c[temp++]=a[p++];
}
//复制
for(int i = first; i<=last; i++)
a[i]=c[i];
}
void MergeSort(node *a,int first,int last)
{
if(first == last)
return;
int mid=(first+last)/2;
MergeSort(a,first,mid);
MergeSort(a,mid+1,last);
Merge(a,first,last);
}
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=0; i<n; i++)
{
cin >> child[i].a;
child[i].b = 0;
}
MergeSort(child,0,n-1);
long long ans = 0;
for(int i=0; i < n ; i++)
{
long long s = child[i].b;
ans += (s+1)*s/2;;
}
cout << ans << endl;
}
return 0;
}