小朋友排队

题目链接

对于每一个位置的高度，如果想要交换次数最少，肯定要把在它后边比它小的换到前面，在它前面比它大的换到后边。然后我们发现其实这就是求整个序列的逆序对数，普通的暴力方法求逆序对肯定会超时，我们采用树状数组求逆序对。然后对于每个位置求得的结果带入等差数列求和公式求得结果。

代码如下：

#include<bits/stdc++.h>
using namespace std;
int n;
int c[100007],d[100007];
long long ans[100007];
struct node
{
   int val;
   int pos;
}p[100007];
int lowbit(int x)
{
    return x&(-x);
}
bool cmp(node x,node y)
{
    return x.val<y.val;
}
void in(int x)
{
    while(x<=n)
    {
        c[x]++;
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    long long sum=0;
    while(x)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&p[i].val);
        p[i].pos=i;
    }
    sort(p+1,p+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        d[p[i].pos]=i;
        if(i!=1&&p[i].val==p[i-1].val)
        d[p[i].pos]=d[p[i-1].pos];//将数据离散化，并且要对于相等的数据特殊处理
    }
    for(int i=1;i<=n;i++)
    {
        in(d[i]);
        ans[i]=i-getsum(d[i]);//求得应该在这个位置前面但在这个位置后边的个数
    }
    memset(c,0,sizeof(c));
    for(int i=n;i>=1;i--)
    {
        in(d[i]);
        ans[i]+=getsum(d[i]-1);//求得应该在这个位置后边但是在这个位置前面的个数
    }
    long long sum=0;
    for(int i=1;i<=n;i++)
    {
        sum+=(ans[i]+1)*ans[i]/2;//等差数列求和公式求和
    }
    printf("%lld\n",sum);
    return 0;
}