解题思路
难点在于问题的转化,最大值等于在3n中选n块不相邻的披萨使总和最大,并且首尾算相邻。
对于不考虑首尾的情况,想得到选n块不相邻的总和最大的方法是dp;
dp[i][j]表示i之前选j块的最大值,对于位置i来说,要么不选,从dp[i-1][j]直接转移来,要么选择,从dp[i-2][j-1]转移而来,所以转移方程为dp[i][j] = max(dp[i-1][j], dp[i - 2][j - 1] + slices[i]);
考虑首尾情况,即去掉首和尾分别进行dp,这样就满足首尾不会同时选中,两次dp的最大值就是答案;
实现上,先处理初始两个位置避免越界。
代码
class Solution {
public:
int maxSizeSlices(vector<int>& slices) {
int all = slices.size();
int n = all / 3;
vector<vector<int>> dp(all, vector<int>(n + 1));
dp[0][1] = slices[0];
dp[1][1] = max(slices[0], slices[1]);
for(int i = 2; i < all - 1; i++) {
for(int j = 1; j <= n; j++) {
dp[i][j] = max(dp[i-1][j], dp[i - 2][j - 1] + slices[i]);
}
}
int ans1 = dp[all-2][n];
dp = vector<vector<int>>(all, vector<int>(n + 1));
dp[1][1] = slices[1];
dp[2][1] = max(slices[1], slices[2]);
for(int i = 3; i < all; i++) {
for(int j = 1; j <= n; j++) {
dp[i][j] = max(dp[i-1][j], dp[i - 2][j - 1] + slices[i]);
}
}
int ans2 = dp[all-1][n];
return max(ans1, ans2);
}
};