UVa 10271. Chopsticks

题意为用n个数据，要从中选出k+8个三元组（x，y，z，其中x<=y<=z)，每选一次的代价为（x-y）^2，求最小代价和。

考虑动态规划，那么很容易想到如下的状态定义：dp[i][j]表示用前j个数组成i个(x,y,z)数对的最小消耗。

 

考虑状态d[i][j]:
对于x和y，有如下考虑：
对于a[j]，如果不使用a[j]，那么d[i][j]=d[i][j-1]；
如果使用a[j]，那么就和a[j-1]一起使用，d[i][j]=d[i-1][j-2]+w(a[i],a[i-1])；
于是有了总的状态转移方程：d[i][j]=min{d[i][j-1],d[i-1][j-2]+w(a[i],a[i-1])}；
这应该不难理解，但是对于z的决策呢？
如果把a[i]按降序排列，那么z的影响就可以忽略了！这样依然可以采用上面的方程。

 

#include <iostream>
#include <cstdio>
#include <string.h>
#include <cmath>
using namespace std;

int dp[1050][5010];
int ch[5010];
int k, n;
int main()
{
	int t;
	scanf("%d", &t);
	while( t--)
	{
		scanf("%d%d", &k, &n);
		int i,j;
		for( i=n-1; i>=0; i--)
			scanf("%d", &ch[i] );
		k=k+8;
		memset( dp, 0x7f, sizeof(dp) );
		for( i=0; i<n; i++) dp[0][i] =0;
		for( i=1; i<=k; i++)
			for( j=i*3-1; j<n; j++)
			{
				dp[i][j] =dp[i][j-1];
				if( dp[i-1][j-2]+ (ch[j]-ch[j-1])*(ch[j]- ch[j-1])< dp[i][j] )
					dp[i][j] =dp[i-1][j-2]+ (ch[j]-ch[j-1])*(ch[j]- ch[j-1]);
			}
		printf("%d\n", dp[k][n-1] );
	}
	return 0;
}