（博客搬迁）pku2231 Moo Volume(数学题http://poj.org/problem?id=2231）

来源：互联网发布：mac air键盘灯不亮编辑：程序博客网时间：2024/05/21 10:33

分析：求所有牛的通信路程的总和。查看数据，如果用二重循环会超时，而且数据比较大，所以要用__int64型。因为这是一维的，所以可以先排序，然后可以推出一个数学关系式。每增加一头牛，路程总和会增加（pos[i] - pos[i-1])*2 + (pos[i] - pos[i-2])*2 + ...+(pos[i] - pos[1])*2 = {(pos[i] * (i-1) - (pos[1] + pos[2] + ... +pos[i-1]) }* 2 ,

code:

#include<stdio.h>
#include<stdlib.h>
#define N 10002
__int64 pos[N];

int cmp(const void *a, const void *b)
{
return *(__int64*)a - *(__int64*)b;
}

int main()
{
__int64 sum;
int i, n;
while(scanf("%d", &n) != EOF)
{
  for(i = 1; i <= n; i++)
      scanf("%I64d", &pos[i]);
  qsort(pos+1, n, sizeof(pos[0]), cmp);
  sum = 0;
  for(i = 2; i <= n; i++)
  {
   sum += pos[i] * (i-1) - pos[i-1];
   pos[i] += pos[i-1];
  }
  sum *=2;
  printf("%I64d\n", sum);
}
}