首页 > 代码库 > POJ 2231 Moo Volume

POJ 2231 Moo Volume

这题用的是贪心算法来算的,贪心策略如下:

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;
}


POJ 2231 Moo Volume