House Robber

代码如下,注释部分为递归版本,在OJ上运行超时,在本地正常,所以选择迭代版本。
代码如下:

#include<iostream>
#include<vector>
using namespace std;

class Solution {
public:
    int rob(vector<int>& nums) {
        int hasFinal, exceptFinal;

        if(nums.size() == 0)
            return 0;

        if(nums.size() == 1)
            return nums[0];

        if(nums.size() == 2) {
            if(nums[0] > nums[1])
                return nums[0];
            else
                return nums[1];
        }

        int tmp1, tmp2;
        int maxNM1 = max(nums[0], nums[1]);
        int maxNM2 = nums[0];
        int maxN;
        for(int i = 3; i <= nums.size(); i++) {
            if(i != 3) {
                maxNM2 = maxNM1;
                maxNM1 = maxN;
            }

            tmp1 = maxNM2 + nums[i - 1];
            tmp2 = maxNM1;
            maxN = max(tmp1, tmp2);
        }

        return maxN;


        //recursive version
        /*int hasFinal, exceptFinal;

        if(nums.size() == 0)
            return 0;

        if(nums.size() == 1)
            return nums[0];

        if(nums.size() == 2) {
            if(nums[0] > nums[1])
                return nums[0];
            else
                return nums[1];
        }

        if(nums.size() == 3) {
            hasFinal = nums[0] + nums[2];
            if(nums[0] > nums[1])
                exceptFinal = nums[0];
            else
                exceptFinal = nums[1];
            if(hasFinal > exceptFinal)
                return hasFinal;
            else
                return exceptFinal;
        }


        //int hasFinal, exceptFinal;
        vector<int> arr1(nums.begin(), nums.end() - 2);
        vector<int> arr2(nums.begin(), nums.end() - 1);
        hasFinal = rob(arr1) + *(nums.end() - 1);
        exceptFinal = rob(arr2);
        if(hasFinal > exceptFinal)
            return hasFinal;
        else
            return exceptFinal;*/
    }
};

int main() {
    vector<int> arr;
    arr.push_back(5);arr.push_back(4);arr.push_back(1);arr.push_back(6);
    arr.push_back(57);arr.push_back(24);arr.push_back(48);arr.push_back(12);
    Solution solution;
    cout << solution.rob(arr) << endl;
    getchar();
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值