UVA10271- Chopsticks

题意：给出n只筷子，要从中挑选出k+3对(x，y，z)，使得所有对的(x - y) ^ 2之和最小

思路：本题跟这题有些一样点击打开链接，但是这题的组合是三个数，而且z要比其他两个都要大。

d[i][j]表示前i只筷子组合成j对(x, y, z)的最小的平方差的和，对于第i只筷子就存在两种情况，取或者不取。取的话arr[i]就要与arr[i - 1]配对，d[i][j] = min(d[i - 1][j], d[i - 2][j - 1]  + (arr[i] - arr[i - 1]) ^ 2);不取的话，d[i][j] = d[i- 1][j]；

如果从大到小开始取筷子的话，就不用担心存在有配对中z不满足的情况。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long ll;
const int MAXN = 5005;
const int INF = 0x3f3f3f3f;

int arr[MAXN];
int K, n;
ll d[MAXN][MAXN];

void init() {
    for (int i = 0; i <= n; i++) {
        d[i][0] = 0;
        for (int j = 1; j <= K; j++) 
            d[i][j] = INF;
    }
}

void dp() {
    for (int i = 3; i <= n; i++)
        for (int j = 1; j <= K; j++) 
            if (i >= j * 3) 
                d[i][j] = min(d[i - 2][j - 1] + (arr[i] - arr[i - 1]) * (arr[i] - arr[i - 1]), d[i - 1][j]);  
}

int main() {
    int cas;    
    scanf("%d", &cas);
    while (cas--) {
        scanf("%d %d", &K, &n); 
        K += 8;
        for (int i = n; i >= 1; i--)
            scanf("%d", &arr[i]);
        init();
        dp();
        printf("%lld\n", d[n][K]); 
    }
    return 0;
}