经典的动态规划问题,首先应该把筷子从大到小来排序,dp[i][j]表示用前j跟筷子构造i个三元组的最小代价,注意j应该从i*3开始,在计算dp[i][j]的时候,考虑第j根筷子,这根筷子要么是第i个三元组中最小的,要么不是第i个三元组中最小的,那么dp[i][j]就是dp[i][j-1]和dp[i-1][j-2]+(stick[j]-stick[j-1])*(stick[j]-stick[j-1]) 这两个数值中较小的一个,其实这样考虑的话,如果第j根筷子参与到了第i个三元组的组成,那么其实在前面的j-2根筷子里面一定可以找到至少一根剩下的不在前面的三元组中的一根筷子,连同第j根和第j-1根筷子组成第i个三元组,这样一来很自然地每个三元组中最长的那根筷子的问题就得到解决了。
#include <stdio.h>
#include <string.h>
int sticks[5005];
int dp[1005][5005];
void func(int set_num, int stick_num){
int i, j, t;
for(i=1, j=stick_num; i<j; i++, j--){
t = sticks[i];
sticks[i] = sticks[j];
sticks[j] = t;
}
for(j=0; j<=stick_num; j++) dp[0][j] = 0;
for(i=1; i<=set_num+8; i++){
for(j=1; j<=stick_num; j++){
if(j < 3*i){
dp[i][j] = -1;
continue;
}
dp[i][j] = -1;
if(dp[i-1][j-2] != -1){
dp[i][j] = dp[i-1][j-2] + (sticks[j]-sticks[j-1])*(sticks[j]-sticks[j-1]);
}
if(dp[i][j-1] != -1){
if(dp[i][j] == -1){
dp[i][j] = dp[i][j-1];
}
else{
if(dp[i][j-1] < dp[i][j])
dp[i][j] = dp[i][j-1];
}
}
}
}
printf("%d\n", dp[set_num+8][stick_num]);
}
int main(void){
int n, set_num, stick_num;
int i;
//freopen("input.dat", "r", stdin);
scanf("%d", &n);
while(n--){
scanf("%d %d", &set_num, &stick_num);
for(i=1; i<=stick_num; i++){
scanf("%d", sticks+i);
}
func(set_num, stick_num);
}
return 0;
}