这题用的是贪心算法来算的,贪心策略如下:
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; }
时间: 2024-10-08 12:45:50