UVA - 10271 Chopsticks 线性DP

最新推荐文章于 2021-11-12 00:58:21 发布

carut

最新推荐文章于 2021-11-12 00:58:21 发布

阅读量112

点赞数

分类专栏： # 紫书第九章动态规划文章标签：算法动态规划

本文链接：https://blog.csdn.net/zpf1998/article/details/104137886

版权

紫书第九章动态规划专栏收录该内容

37 篇文章 1 订阅

订阅专栏

问题

在N个数字中挑选K组，每组3个数字，A<=B<=C，要求 $\sum_i(A_i-B_i)^2$ 最小

分析

除了最长的筷子，其他两只短的一定是相邻的
dp[i][j]，前i只筷子分给j个人的最小代价
状态转移方程：
$i < 3 * j ， d p [i] [j] = I N F$
$i > = 3 * j$ , 第i个筷子分给j的代价 $d p [i - 2] [j] + (l e n [i - 1] - l e n [i]) * (l e n [i - 1] - l e n [i]) ，不分给 j 的代价 d p [i - 1] [j]$ ，所以
$d p [i] [j] = m i n (d p [i - 1] [j], d p [i - 2] [j - 1] + (l e n [i - 1] - l e n [i]) * (l e n [i - 1] - l e n [i]))$

#include<iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
using namespace std;
const int maxk=1020,maxc=5005,maxp=55,INF=0x3f3f3f3f;
int T,k,n,len[maxc],dp[maxc][maxk]; //dp[i][j]，前i只筷子分给j个人的最小代价
//除了最长的筷子，其他两只短的一定是相邻的
int main(void){
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&k,&n);
        k+=8;
        for(int i=n;i>0;--i){
            scanf("%d",&len[i]);  //逆序排列，从大到小
        }
        for(int i=1;i<=n;++i){
            for(int j=k;j>=1;--j){
                if(i<3*j) dp[i][j]=INF;
                else break;
            }
        }

        for(int i=1;i<=n;++i){
            for(int j=1;j<=k;++j){
                //保持足够的筷子分配
                if(i>=3*j){
                    //第i个筷子不分给j dp[i-1][j]，分给了j  dp[i-2][j]+(len[i-1]-len[i])*(len[i-1]-len[i])
                    dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(len[i-1]-len[i])*(len[i-1]-len[i]));
                }
            }
        }
        printf("%d\n",dp[n][k]);
    }
    return 0;
}