人丑就要多读书的博客

Brevity is the soul of wit

leetcode198-House Robber

题目:

You are a professional robber planning to rob houses along a street. Each house has a certain amount of money stashed, the only constraint stopping you from robbing each of them is that adjacent houses have security system connected and it will automatically contact the police if two adjacent houses were broken into on the same night.

Given a list of non-negative integers representing the amount of money of each house, determine the maximum amount of money you can rob tonight without alerting the police.

链接:https://leetcode.com/problems/house-robber/

法一:状态压缩

这道理可以看做是状态压缩,每两个数字看做是一行,状态有3个,故需要F[N][3]的数组,F[i][j]就表示第i行状态j时rob的money。

具体状态压缩可以看我这两篇blog: 算法练习系列—hiho1048 状态压缩一(铺地砖)  算法练习系列—hiho1044 状态压缩二(捡垃圾)

  1. #include <iostream>  
  2. #include <algorithm>  
  3. #include <vector>  
  4. using namespace std;  
  5.   
  6. #define N 10000  
  7.   
  8. class Solution {  
  9. public:  
  10.     Solution(){  
  11.         memset(F, 0, sizeof(int)*N*3);  
  12.     }  
  13.     int rob(vector<int> &num) {  
  14.         if(num.size() == 0) return 0;  
  15.         else if(num.size() == 1) return num[0];  
  16.         else{  
  17.             // 初始化第一行  
  18.             F[0][0] = 0;  
  19.             F[0][1] = num[1];  
  20.             F[0][2] = num[0];  
  21.             int maxMoney = max(num[1], num[0]);  
  22.             for(int i = 1; i < num.size()-1; i++){  
  23.                 for(int j = 0; j < 3; j++){       
  24.                     int t = j >> 1;  
  25.                     for(int k = 0; k < 3; k++){  
  26.                         if( t ==  (k & (1>>0))){      // 判断第i行状态j与第i-1行各个状态是否兼容 即j的倒数第二位与k的倒数第一位是否相同  
  27.                             F[i][j] = max(F[i-1][k], F[i][j]);  //  输出兼容的最大者  
  28.                         }  
  29.                     }  
  30.                     if((j & (1<<0)) == 1) F[i][j] = F[i][j] + num[i+1];  // 如果状态j最后一位为1则 需要加上该方案数  
  31.                     if(F[i][j] > maxMoney){                // 求最大的方案数  
  32.                         maxMoney = F[i][j];  
  33.                     }  
  34.                 }  
  35.             }  
  36.             return maxMoney;  
  37.         }  
  38.     }  
  39. private:  
  40.     int F[N][3];  
  41. };  

法二:动态规划

这题可以看做是简单的DP问题,用A[0]表示没有rob当前house的最大money,A[1]表示rob了当前house的最大money,那么A[0] 等于rob或者没有rob上一次house的最大值

即A[i+1][0] = max(A[i][0], A[i][1])..  那么rob当前的house,只能等于上次没有rob的+money[i+1], 则A[i+1][1] = A[i][0]+money[i+1].

实际上只需要两个变量保存结果就可以了,不需要用二维数组

  1. class Solution {  
  2. public:  
  3.     int rob(vector<int> &num) {  
  4.         int best0 = 0;   // 表示没有选择当前houses  
  5.         int best1 = 0;   // 表示选择了当前houses  
  6.         for(int i = 0; i < num.size(); i++){  
  7.             int temp = best0;  
  8.             best0 = max(best0, best1); // 没有选择当前houses,那么它等于上次选择了或没选择的最大值  
  9.             best1 = temp + num[i]; // 选择了当前houses,值只能等于上次没选择的+当前houses的money  
  10.         }  
  11.         return max(best0, best1);  
  12.     }  
  13. };  
阅读更多
文章标签: leetcode 198
个人分类: leetcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

leetcode198-House Robber

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭