为了使花费最小,每次一定是选取大小最接近的一组物品,所以先对所有的物品按大小排序,每个物品的选择范围就是左右两个物品,所以我们可以选定一个物品后,只绑定到它右侧的物品,然后dp过程中只涉及到当前物品选不选的两种状态变化走向,所以得出以下dp方程:
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])
代码如下:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define MAX 2007
using namespace std;
int dp[MAX][MAX];
int a[MAX];
int n,k;
int main ( )
{
while ( ~scanf ( "%d%d" , &n , &k ) )
{
for ( int i = 1 ; i <= n ; i++ ) scanf ( "%d" , &a[i] );
sort ( a+1 , a+n+1 );
memset ( dp , 0x3f , sizeof ( dp ) );
dp[0][0] = 0;
dp[1][0] = 0;
for ( int i = 2 ; i <= n ; i++ )
for ( int j = 0 ; j <= k && j*2<=i ; j++ )
{
dp[i][j] = min ( dp[i][j] , dp[i-1][j] );
if ( j > 0 ) dp[i][j] = min ( dp[i-2][j-1] + (a[i]-a[i-1])*(a[i]-a[i-1]), dp[i][j]);
}
printf ( "%d\n" , dp[n][k] );
}
}