leetcode-213

最新推荐文章于 2024-06-30 22:10:28 发布

cug_coffee

最新推荐文章于 2024-06-30 22:10:28 发布

阅读量145

点赞数

CC 4.0 BY-SA版权

分类专栏： c++基础

本文链接：https://blog.csdn.net/cug_coffee/article/details/109790116

c++基础专栏收录该内容

12 篇文章

订阅专栏

博客提及直接给出代码，强调是最典型的动态规划，指出写出动态规划方程即可。围绕信息技术领域的动态规划展开。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接上代码了。
最典型的动态规划，写出动态规划的方程就好。

这里分两种情况，
0号位选取时，0号位不选取时，len<=3时特殊处理。
f[i] = f[i-2] > f[i-3] ? f[i-2]+a[i] : f[i-3]+a[i];

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>

int rob(std::vector<int>& nums) {
  int res = 0;
  int len = nums.size();
  if (len <= 3) {
    for (int i = 0; i < len; i++) {
      if (nums[i] > res) res = nums[i];
    }
    return res;
  }
  std::vector<int> odd_res(len, 0);
  std::vector<int> even_res(len, 0);
  int i = 0;
  // odd
  odd_res[0] = 0;
  odd_res[1] = nums[1];
  odd_res[2] = nums[2] > nums[1] ? nums[2] : nums[1];
  for (i = 3; i < len; i++) {
    int val = odd_res[i-2] > odd_res[i-3] ? odd_res[i-2] : odd_res[i-3];
    odd_res[i] = val + nums[i];
  }
  for (const auto& e : odd_res) {
    if (res < e) res = e;
  }
  // even
  even_res[0] = nums[0];
  even_res[1] = 0;
  even_res[2] = even_res[0] + nums[2];
  for (i = 3; i < len-1; i++) {
    int val = even_res[i-2] > even_res[i-3] ? even_res[i-2] : even_res[i-3];
    even_res[i] = val + nums[i];
  }
  for (const auto& e : even_res) {
    if (res < e) res = e;
  }
  return res;
}


int main(int argc, char* argv[]) {
  // int arr[] = {10,9,2,5,3,7,101,11,18};
  int arr[] = {2,3,2,2};
  std::vector<int> nums;
  int len = sizeof(arr)/sizeof(int);
  for (int i = 0; i < len; i++) {
    nums.push_back(arr[i]);
  }
  int res = rob(nums);
  std::cout << "the sum is: " << res << std::endl;
  return 0;
}