leetcode 198打家劫舍

时间:2023-03-09 00:51:48
leetcode 198打家劫舍

讲解视频见刘宇波leetcode动态规划第三个视频

记忆化搜索代码:

#include <bits/stdc++.h>
using namespace std;
class Solution {
private:
vector<int>memo;
int tryRob(int index, vector<int>& nums)
{
if (index > nums.size())
{
return ;
}
if (memo[index] != -)
return memo[index];
int i;
int res = ;
for (i = index; i < nums.size(); i++)
{
res = max(res, nums[i] + tryRob(i + , nums));
}
memo[index] = res;
return res;
}
public:
int rob(vector<int>& nums) {
memo = vector<int>(nums.size()+, -);
return tryRob(,nums);
}
};

动态规划代码:

#include<bits/stdc++.h>
#include <algorithm>
using namespace std;
class Solution {
private:
/*bool cmp(int a, int b)
{
return a>b;
}*/
public: int rob(vector<int>& nums) {
int n = nums.size();
vector<int>memo(n + , );
if (n == )
return ;
memo[n] = ;
memo[n+]=;
memo[n - ] = nums[n - ];
memo[n-]=nums[n-];
int i;
int j;
for (i = n - ; i >= ; i--)
{
memo[i] = nums[i];
for(j=i+;j<n;j++)
{
memo[i]=max(memo[i],nums[i]+memo[j]);
} }
sort(memo.begin(),memo.end());
return memo[memo.size()-]; }
};