今日份题目:
给你一个整数数组 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
题目思路
使用动态规划,记元素x在数组中出现的次数为c,我们可以用一个数组sum记录数组nums中所有相同元素之和,即sum[x]=x⋅c。若选择了x,则可以获取sum[x] 的点数,且无法再选择x−1和x+1。因为,选了sum[x]就等于选了x、x-1和x+1三个数,所以不能再选相邻的两个,和打家劫舍一样,不能选相邻的数。
代码
class Solution
{
public:
int deleteAndEarn(vector<int> &nums)
{
int maxn = 0;
for (int i=0;i<nums.size();i++) //找到最大的数字
{
maxn = max(maxn, nums[i]);
}
int sum[30000]={0};
for (int i=0;i<nums.size();i++) //对数组中出现的每个数求其在数组中的总和
{
sum[nums[i]] += nums[i];
}
//动态规划过程
//记元素x在数组中出现的次数为c,我们可以用一个数组sum记录数组nums中所有相同元素之和,即sum[x]=x⋅c。若选择了x,则可以获取sum[x] 的点数,且无法再选择x−1和x+1。因为,选了sum[x]就等于选了x、x-1和x+1三个数,所以不能再选相邻的两个,和打家劫舍一样,不能选相邻的数。
int dp[30000]={0};
dp[0] = sum[0], dp[1] = max(sum[0], sum[1]);
for (int i = 2; i <= maxn; i++)
{
dp[i] = max(dp[i-2] + sum[i], dp[i-1]);//状态转移方程
}
return dp[maxn];
}
};