UVA10271
首先,将筷子逆序读入,即使得读入的
C
h
o
p
s
t
i
c
k
s
Chopsticks
Chopsticks数组中的元素时降序的。这样在考虑前
i
i
i只筷子时可以选
i
i
i和
i
−
1
i-1
i−1,而第三只(最大那只)在
i
−
1
i-1
i−1之前选一个没被用过的即可。
定义:
d
p
[
i
]
[
j
]
dp[i][j]
dp[i][j]为考虑前i个筷子并选了j个三元组后的最小权值和。
初始化:
d
p
[
i
]
[
0
]
=
0
dp[i][0]=0
dp[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
i
i和
i
−
1
i-1
i−1只作为
a
a
a和
b
b
b。
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;
}