LeetCode 打家劫舍+删除并获得点数

题目描述

打家劫舍题目传送门1
删除并获得点数传送门2

思路

这两道题看似毫无关系，但竟然可以用桶数组联系起来！！
先说打家劫舍这道题
限制条件是不能走相邻的屋，再联想到跳台阶（走一格或两格），这其实就是在跳台阶的基础上加了一个限制条件
设f[i]表示打劫到第i家已经获得的最大钱数,则对第i家分为两种情况处理，打劫or不打劫
如果打劫，则他的i-1肯定没法打劫，就相当于是f[i-2]+cost[i]
如果不打劫，就相当于是f[i-1]
则得到了状态转移方程f[i] = max(f[i-2]+cost[i], f[i-1])

再说删除并获得点数题
审题关键是删除nums[i]-1与nums[i]+1的所有数,是不是很像打家劫舍那道题的不能去相邻的屋子进行抢劫，则可以用桶数组，记录每个数字出现的价值，见代码

代码

注意vector是从0开始存的！！！

打家劫舍

//要能看出来这是一个跳台阶问题
const int N = 1e2+10;
int f[N];//表示到第i家偷窃的金额数，若第i偷窃，则加上第i-2家，若没偷窃，加上第i-1家
class Solution {
public:
    int rob(vector<int>& nums) {
        //nums下标是从第零号开始的！！所以状态转移的时候nums下标得往前一位
        f[0] = 0;
        f[1] = nums[0];  //带入f[N]表示的含义去理解，到第1家偷窃的金额数最大，只有这一家，那肯定是偷
        for(int i = 2; i <= nums.size(); i++)
        {
            f[i] = max(f[i-2] + nums[i-1] ,f[i-1]);
        }
        return f[nums.size()];
    }
};

删除并获得点数

这个桶数组并不是仅仅用来排序的，是用来统计价值的，以及边界注意一下！！

//表示相邻点数：用桶数组排序就有相邻的了，把一样的加起来，就变成了打劫的那个题了
const int N = 2e4+10;
int f[N];
int cnt[N];
class Solution {
public:
    int deleteAndEarn(vector<int>& nums) {
    //认真读题，是删除所有点数！！
    memset(cnt, 0, sizeof cnt);
    int n = nums.size();
    for(int i = 0; i < n; i++)
        cnt[nums[i]]+= nums[i];
    for(int i = 0; i <= 4; i++)
        cout<<cnt[i]<<endl;
    //如果等于零直接跳过，或者加上也没啥
    f[0] = cnt[0];
    f[1] = cnt[1];
    for(int i = 2; i <= N-1; i++)  //数可能很大，并不是一直到数组长度！
    {
        f[i] = max(f[i-2]+cnt[i], f[i-1]);
        // cout<<f[i]<<endl;
    }
    return f[N-1];
    }
};