POJ 2231 Moo Volume(dp)

Description
给定n个数字,问任意两数之间的差的总和是多少
Input
第一行为数列长度n(1<=N<=10,000),之后n行每行一个整数
Output
输出任意两数差值总和
Sample Input
5
1
5
3
2
4
Sample Output
40
Solution
先从小到大排序,然后dp转移方程:
dp[i]=dp[i-1] + i*(a[i]-a[i-1])
最后结果是ans=(dp[0]+dp[1]+…+dp[n-1])*2(因为噪声计算是双向的)
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF (1<<29)
#define maxn 10005
typedef long long ll;
int a[maxn];
ll dp[maxn];
int n;
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    sort(a,a+n);
    for(int i=1;i<n;i++)
        dp[i]=dp[i-1]+i*(ll)(a[i]-a[i-1]);
    ll ans=0;
    for(int i=0;i<n;i++)
        ans+=dp[i];
    printf("%lld\n",2*ans);
    return 0;
}

你可能感兴趣的:(POJ 2231 Moo Volume(dp))