天天写算法之Cow Sorting

点击打开链接

这个题，主要是思路问题，能迁移到树状数组真的很不容易

其实使用了两个数组，一个记录每个位置数据是否已经有了，应一个记录数值大小。然后通过一些数学推导，最后得出答案，

每次添加一个数，就判断已有的数据里面，比自己大的有几个，然后把他移动到这些数据之前，因为每次移动的的代价是交换的两个数据的和，所以，就需要写出一个公式来，用到树形数组求和的一步，就是这个地方，很巧妙，全部的sum-当前这个值的sum，就是大于这个数的sum。

#include <iostream>
#include<cstdio>
#include<string.h>
#include<cmath>
using namespace std;
#define MAX 111111
#define repf(i,from,to) for(int i =from ; i<=to ; i++)
#define ll long long
ll c[MAX];//存储树形数组的节点
int num[MAX];//存储每个节点处数的数量
ll a[MAX],ans;//存储输入的数值
int N ;
int logic(int x)
{
    return x&(-x);
}
void Add(int position,int d)
{
    for(int i = position ; i <= N ; i +=logic(i) )
    {
        c[i]+=d;
        num[i]+=1;
    }
}
int getCount(int x)
{
    int res = 0;
    for(int i = x ; i>0 ; i -=logic(i))
    {
        res += num[i];
    }
    return res ;
}

ll getSum(int x){
    ll res = 0 ;
    for(ll i = x ; i >0 ; i -=logic(i))//一开始把i-=logic（x）了，我的天，找了一个小时的bug
    {
        res +=c[i];
    }
    return res ;
}

int main(){
    while(scanf("%d",&N)!=-1)
    {
        memset(a,0,sizeof(a));
        memset(c,0,sizeof(c));
        memset(num,0,sizeof(num));
            ans = 0;
        repf(i,1,N)
        {
                scanf("%d",&a[i]);
                Add(a[i],a[i]);
                int lessThan ,moreThan;
                lessThan =  getCount(a[i]);
                moreThan = i - lessThan;
                if(moreThan!=0)
                {
                    ans+=a[i]*moreThan;
                    ans+=getSum(N)-getSum(a[i]);
                }
        }
        printf("%I64d\n",ans);
    }
}