uva 10271 Chopsticks （DP）

uva 10271 ChopsticksA

题目大意：给出客人数K和筷子数量N（支），总人数M=客人数K + 8（还有家人）。要找出M组筷子，每组包括3支筷子，每组都有一个badness值，badness值为每组中较小的两支筷子的差值。求怎样分组，才能使总差值最小。

解题思路：dp[i][j]表示使用第i支筷子，并已组成j组是的最小badness总值。在取筷子时，要注意在所有未被使用的筷子中，要有比选取的两根筷子更长的筷子。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#define N 5005
#define M 1005
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f;
int num[N], dp[N][M];
int main() {
    int T;
    scanf("%d", &T);
        while (T--) {
            int k, n;
            for (int i = 0; i < N; i++) {
                for (int j = 0; j < M; j++) {
                    dp[i][j] = INF;
                }
            }
            scanf("%d %d", &k, &n);
            k += 8;
            for (int i = 1; i <= n; i++) {
                scanf("%d", &num[i]);
                dp[i][0] = 0;
            }
            dp[0][0] = 0;
            for (int i = n - 2; i >= 1; i--) {
                for (int j = k; j >= 1; j--) {
                    dp[i][j] = dp[i + 1][j];
                    if (dp[i + 2][j - 1] != INF && n - i - j * 3 >= -1) {
                        dp[i][j] = min(dp[i][j], dp[i + 2][j - 1] + (num[i] - num[i + 1]) * (num[i] - num[i + 1]));
                    }
                }
            }
            printf("%d\n", dp[1][k]);
        }
    return 0;
}