这题用的是贪心算法来算的,贪心策略如下:
dist数组表示各个牛的位置距第一个牛的距离,当然之前要对牛的位置进行升序排序。
设a为第一头牛距各头牛的距离的总和,然后从第二头牛开始遍历,假设遍历到第i头牛时,那么标准为:
t=t-(n-i)*d+i*d,
各变量的含义为:t一开始为a,然后每次就是该表达式运算之后的值;
n为牛的总数;
d为第i头牛和第i-1头牛之间的距离。这个公式画图其实很好理解的。
把所有的t值加起来就是最后的结果。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
using namespace std;
const int N=10005;
int dist[N],cow[N];
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
scanf("%d",&cow[i]);
sort(cow,cow+n);
long long ans=0;
for(int i=0;i<n;i++)
{
dist[i]=cow[i]-cow[0];
ans+=dist[i];
}
long long temp=ans;
for(int i=1;i<n;i++)
{
int d=dist[i]-dist[i-1];
temp=temp-(n-i)*d+i*d;
//printf("%lld\n",temp);
ans+=temp;
}
printf("%lld\n",ans);
}
return 0;
}