首页 > 代码库 > uva 10271 Chopsticks 【dp】

uva 10271 Chopsticks 【dp】

题目:uva 10271 Chopsticks


题意:从一组数中选出每三个为一组,价值为三个中两个小的差值的平方和,让这个总价值最小。


分析:定义dp【i】【j】为从后 i 个中选出 j 对的最小价值。


转移方程:dp【i】【j】 = min(dp【i-1】【j】,dp【i+2】【j-1】+(a【i】-a【i+1】)*(a【i】-a【i+1】))


注意状态转移的条件:就是每组要保留一个最大的值作为第三个值。


AC代码:

 
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <queue>
typedef long long ll;
using namespace std;
const int inf = 0x3f3f3f3f;
const int N = 5500;
int a[N];
int dp[N][2000];
int cmp(int x,int y)
{
    return x>y;
}
int main()
{
    int pro,n;
    while(~scanf("%d%d",&pro,&n))
    {
        memset(dp,inf,sizeof(dp));
        for(int i=n; i>=1; i--)
            scanf("%d",&a[i]),dp[i][0]=0;
        sort(a+1,a+n+1,cmp);
        for(int i=3;i<=n;i++)
        {
            for(int j=1;j<=min(pro,i/3);j++)
                    dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
        }
        printf("%d\n",dp[n][pro]);
    }
    return 0;
}        


uva 10271 Chopsticks 【dp】