leetcode-198强盗抢劫(动态规划)

题目:你是一个专业的强盗,计划在街上抢劫房屋。 每个房子都藏着一定数量的钱,阻止你抢劫他们的唯一限制因素是相邻的房屋有连接的安全系统,如果两个相邻的房子在同一个晚上被闯入,它将自动联系警方。给出一个代表每个房子的金额的非负整数列表,确定今晚可以抢劫的最大金额而不警告警察。

Input: [2,7,9,3,1]
Output: 12
Explanation: Rob house 1 (money = 2), rob house 3 (money = 9) and rob house 5 (money = 1).
             Total amount you can rob = 2 + 9 + 1 = 12.

分析:不能取相邻元素,大问题转化为小问题。对每个元素进行判断,并存储当前元素下最大和。

如果输入为空,返回0;如果输入个数为1,返回这个数;如果说如个数为2,返回最大值。

输入个数大于等于3,就需要运用动态规划。将当前数字和往前数两个(因为是隔一个取)的数字和相加,并和往前数一个的数字和进行对比,取最大。dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);

 

class Solution {
public:
    int rob(vector& nums) {
        int size = nums.size();
	if (size == 0) return 0;
	if (size == 1) return nums[0];

	vector dp(size, 0);
	dp[0] = nums[0];
	dp[1] = nums[0]>nums[1] ? nums[0] : nums[1];
	for (int i = 2; i < size; ++i){
		dp[i] = max(dp[i - 2] + nums[i], dp[i - 1]);
	}
	return dp[size - 1];
    }
};

 

你可能感兴趣的:(C++,动态规划)