给你一个整数数组 nums ,你可以对它进行一些操作。
每次操作中,选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
示例 1:
输入:nums = [3,4,2]
输出:6
解释:
删除 4 获得 4 个点数,因此 3 也被删除。
之后,删除 2 获得 2 个点数。总共获得 6 个点数。
示例 2:
输入:nums = [2,2,3,3,3,4]
输出:9
解释:
删除 3 获得 3 个点数,接着要删除两个 2 和 4 。
之后,再次删除 3 获得 3 个点数,再次删除 3 获得 3 个点数。
总共获得 9 个点数。
提示:
1 <= nums.length <= 2 * 104
1 <= nums[i] <= 104
感觉有点小难,这是我第一遍写的
class Solution {
private:
int rob(vector<int>& nums) {
if(nums.size()==1)
{
return nums[0];
}
int first = nums[0];
int second = max(nums[0],nums[1]);
for(int i=2;i<nums.size();i++)
{
int temp = second;
second = max(first+nums[i],second);
first = temp;
}
return second;
}
public:
int deleteAndEarn(vector<int>& nums) {
sort(nums.begin(),nums.end());
return rob(nums);
}
};
排序之后其实没有完成题目要求,但是过了俩样例,其实题目要求不是真的删掉i-1和i+1的元素,通过转换问题会比较好思考点,223334,很难想象到完成题目的方式是把相同的数字累加起来,再调用打家劫舍的方法换成4 9 4这样,直接取到9,这需要创建
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
private:
int rob(vector<int>& nums) {
if(nums.size()==1)
{
return nums[0];
}
int first = nums[0];
int second = max(nums[0],nums[1]);
for(int i=2;i<nums.size();i++)
{
int temp = second;
second = max(first+nums[i],second);
first = temp;
}
return second;
}
public:
int deleteAndEarn(vector<int>& nums) {
if(nums.size()==0)
{
return 0;
}
int maxVal = *max_element(nums.begin(),nums.end());//找到最大值
vector<int> sum(maxVal+1,0);//创建一个动态结构顺序表,大小是maxval+1,并且将所有的元素都初始化为0,用于储存每个数字出现的累加值
for(int i=0; i<nums.size();i++)
{
int num=nums[i]; //将nums[i]的值赋给num
sum[num] += num; //将num的值加到sum[num]上
}
return rob(sum);
}
};
int main()
{
vector<int> nums = {2,2,3,3,3,4};
Solution s;
cout<<s.deleteAndEarn(nums)<<endl;
return 0;
}
一个新的容器来储存累加后的数,再调用rob就可以输出正确答案