class Solution {
public:
int tallestBillboard(vector<int>& rods) {
//暴力:3^n
//即分三组:1 -1 0,使得和为0的前提下正数和最大
//类似目标和,但是目标和sum找不到关系
//dp[i][左边和][右边和]的话 20*5000*5000空间上就炸了,考虑状态中记录差绝对值,最大高度
/*状态:dp[i][高低两边的差值绝对值]=高低两组公共高度的最大值
不放:dp[i][j]=dp[i-1][j]
放高的一边:dp[i][j+rods[i]]=max(dp[i-1][j],dp[i-1][j+rods[i]])
放低的一边:dp[i][abs(rods[i]-j)]=max(dp[i-1][j]+min(rods[i],j),dp[i-1][abs(rods[i]-j)])
ans:dp[n-1][0]
初始化:dp[-1][0...5001]都是0
*/
int n = rods.size(), sum = accumulate(rods.begin(), rods.end(), 0);
vector<vector<int>> dp(n + 1, vector<int>(sum + 1));
for (int i = 0; i < n; ++i) {
for (int j = 0; j <= sum; ++j) {
dp[i + 1][j] = max(dp[i + 1][j], dp[i][j]);
if (j + rods[i] <= sum) {
dp[i + 1][j + rods[i]] = max(dp[i][j], dp[i][j + rods[i]]);
}
dp[i + 1][abs(rods[i] - j)] = max(dp[i][j] + min(rods[i], j), dp[i][abs(rods[i] - j)]);
}
}
return dp[n][0];
}
};
leetcode956.最高的广告牌(困难)
最新推荐文章于 2022-07-02 23:52:07 发布