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];}
};