hdu 1421 dp

最新推荐文章于 2020-04-02 21:00:00 发布

原创最新推荐文章于 2020-04-02 21:00:00 发布 · 360 阅读

0 ·

CC 4.0 BY-SA版权

本文通过使用C++实现了一个动态规划算法来解决给定数组中如何将数组分成K个连续子数组使得相邻子数组间差异最小的问题。该算法首先对输入数组进行排序，然后运用动态规划求解最优解。

#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
#define inf 1000000000;
using namespace std;
int d[2005][2005],a[2005];
int main()
{   int n,k,i,j;
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
    while(scanf("%d%d",&n,&k)!=EOF){
            for(i=1;i<=n;i++)
                scanf("%d",&a[i]);
    sort(a+1,a+n+1);
 
    for(i=0;i<=n;i++)
        for(j=1;j<=k;j++)
        d[i][j]=inf;
    d[0][0]=0;
    for(i=2;i<=n;i++)
        for(j=1;j<=k;j++)
        {
            d[i][j]=min(d[i-1][j],d[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
        }

    printf("%d\n",d[n][k]);
}
    return 0;
}