
UVA10271
首先,将筷子逆序读入,即使得读入的C h o p s t i c k s ChopsticksChopsticks数组中的元素时降序的。这样在考虑前i ii只筷子时可以选i ii和i − 1 i-1i−1,而第三只(最大那只)在i − 1 i-1i−1之前选一个没被用过的即可。
定义:
d p [ i ] [ j ] dp[i][j]dp[i][j]为考虑前i个筷子并选了j个三元组后的最小权值和。
初始化:
d p [ i ] [ 0 ] = 0 dp[i][0]=0dp[i][0]=0
不选择任何筷子权值和当然是0
转移方程:
d p [ i ] [ j ] = m i n ( d p [ i − 1 ] [ j ] , d p [ i − 2 ] [ j − 1 ] + S q u a r e ( C h o p s t i c k s [ i − 1 ] − C h o p s t i c k s [ i ] ) ) ; dp[i][j] = min(dp[i - 1][j],dp[i - 2][j - 1] + Square(Chopsticks[i - 1] - Chopsticks[i]));dp[i][j]=min(dp[i−1][j],dp[i−2][j−1]+Square(Chopsticks[i−1]−Chopsticks[i]));决策为不选或者选择第i ii和i − 1 i-1i−1只作为a aa和b bb。
AC代码:
#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
#include<map>
using namespace std;
constexpr static int inf = 0x3f3f3f3f;
int Chopsticks[5001];
int K, N;
int dp[5001][1009];
void Input() {
cin >> K >> N;
for (int i = N; i >= 1; --i) {
cin >> Chopsticks[i];
}
K += 8;
}
int Square(const int& x) {
return x * x;
}
int DP() {
memset(dp, 0x3f, sizeof(dp));
for (int i = 1; i <= N; ++i) {
dp[i][0] = 0;
}
for (int i = 3; i <= N; ++i) {
for (int j = 1; j <= K; ++j) {
//筷子不够组成j个三元组
if (i < 3 * j) {
break;
}
dp[i][j] = min(
dp[i - 1][j],
dp[i - 2][j - 1] + Square(Chopsticks[i - 1] - Chopsticks[i])
);
}
}
return dp[N][K];
}
int main() {
ios::sync_with_stdio(false);
int T;
cin >> T;
while (T--) {
Input();
cout << DP() << endl;
}
return 0;
}版权声明:本文为qq_42971794原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。