C++ 代码实现房屋盗窃问题:最大化收益,避免触发警报
以下是使用 C++ 编写的完整代码,可以计算在不触动警报装置的情况下,一夜之内能够偷窃到的最高金额。
#include <iostream>
#include <vector>
#include <algorithm>
int rob(std::vector<int>& nums) {
int n = nums.size();
if (n == 0) return 0;
if (n == 1) return nums[0];
std::vector<int> dp(n, 0);
dp[0] = nums[0];
dp[1] = std::max(nums[0], nums[1]);
for (int i = 2; i < n; i++) {
dp[i] = std::max(dp[i-1], dp[i-2] + nums[i]);
}
return dp[n-1];
}
int main() {
std::vector<int> nums = {1, 2, 3, 1};
int result = rob(nums);
std::cout << "最高金额: " << result << std::endl;
nums = {2, 7, 9, 3, 1};
result = rob(nums);
std::cout << "最高金额: " << result << std::endl;
return 0;
}
这段代码使用动态规划的思想解决了问题。定义一个大小为n的数组dp,其中dp[i]表示从0到第i个房屋能够偷窃到的最高金额。对于每个房屋,可以选择偷窃或者不偷窃。如果选择偷窃第i个房屋,则最高金额为dp[i-2] + nums[i];如果选择不偷窃第i个房屋,则最高金额为dp[i-1]。因此,状态转移方程为dp[i] = max(dp[i-1], dp[i-2] + nums[i])。最终的结果即为dp[n-1],其中n为房屋的个数。
代码中给出了两个示例的使用方法,你可以根据需要修改输入数组nums以测试其他情况。运行代码后,会输出最高金额的结果。
原文地址: https://www.cveoy.top/t/topic/cxxs 著作权归作者所有。请勿转载和采集!