题目描述
打家劫舍题目传送门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];
}
};