题目
https://leetcode-cn.com/problems/pizza-with-3n-slices
不是贪心,如果选择了最大的,但两边删去的加起来比原来的还大。
选择了一个位置后,剩下的位置除了邻近的都可以选择
选择n/3个位置。这是第二个限制,如果剩下的位置除了邻近的都可以选择,最优选择是n/2个。这也是动态规划需要记录已选择个数。
开头需要枚举。因为开始会删去已选择位置的左右,也是一个环,所以第一个位置选择了会删去最后一个,所以不能考虑最后一个位置。若是没有选择第一个位置,则不影响最后一个位置,可以考虑最后一个位置。
动态规划状态定义:
dp[位置][已选择个数],考虑这个位置及之前的位置、选择了这么多个数的状态。
转移:
遍历每一个位置,如果不选择这个位置,则可以取前一个位置的状态。最优值在这里传递,对应了状态定义里的考虑这个位置之前。
如果选择这个位置,则要取前2个位置的状态,并且已选取位置少一个。注意这里前一个位置是用来前两个位置选择删去的,这个位置左边删去的位置在更左的地方,因为是一个环,包括末尾,已选择个数不超过n/3保证肯定可以删、后面不会选择已经删掉的位置。
答案选择
根据dp状态定义,选择dp[n-2][n/3],dp[n-1][n/3],是否考虑开头。
class Solution {
public:
int dp[505][500/3+5];
int maxSizeSlices(vector<int>& slices) {
memset(dp,0,sizeof(dp));
int n=slices.size();
dp[0][1]=slices[0];
int ans=0;
for(int i=1;i<n-1;i++){
for(int j=1;j<=n/3 ;j++){
if(i-2<0|| j-1<0)
dp[i][j]=max(slices[i],dp[i-1][j]);
else
dp[i][j]=max(dp[i-2][j-1]+slices[i],dp[i-1][j]);
}
}
ans=dp[n-2][n/3];
memset(dp,0,sizeof(dp));
for(int i=1;i<n;i++){
for(int j=1;j<=n/3 ;j++){
if(i-2<0|| j-1<0)
dp[i][j]=max(slices[i],dp[i-1][j]);
else
dp[i][j]=max(dp[i-2][j-1]+slices[i],dp[i-1][j]);
}
}
ans=max(ans,dp[n-1][n/3]);
return ans;
}
};