leetcode 198. House Robber

最新推荐文章于 2024-01-18 14:02:27 发布

weixin_34219944

最新推荐文章于 2024-01-18 14:02:27 发布

阅读量45

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/willaty/p/8384517.html

本文介绍了解决整数集合中选取非相邻元素以获得最大和的问题，提供了两种解决方案：一种是使用动态规划的方法，另一种是通过计算差值来确定是否选择当前元素。这两种方法都能有效地解决该问题。

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

给定一系列整数，取整数，相邻不能取。

解法有二：

动态规划：

    int rob(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        if (nums.size() == 1) return nums[0];
        if (nums.size() == 2) return max(nums[0], nums[1]);
        int ret = 0;
        int last_2 = nums[0];
        int last_1 = max(nums[1], nums[0]);
        
        for (int i = 2; i < nums.size(); ++i) {
            ret = max(last_2 + nums[i], last_1);
            last_2 = last_1;
            last_1 = ret;
        }
        return ret;
    }

求差：

取last = nk - n(k - 1) + n(k - 2) - n(k - 3)...，当遍历到当前元素时，做差，如果结果为正数，则取；否则置0（相当于断开）。

而上式和sum求和除二等于相邻元素和。

    int rob(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        int last = 0;
        int sum = 0;
        int ret = 0;
        
        for (int i = 0; i < nums.size(); ++i) {
            if (nums[i] - last <= 0) {
                ret = ret + (last + sum) / 2;
                last = 0;
                sum = 0;
            } else {
                last = nums[i] - last;
                sum += nums[i];
            }
        }
        ret = ret + (last + sum) / 2;
        
        return ret;
    }