leetcode 740. 删除并获得点数(dp)

最新推荐文章于 2024-09-24 08:39:28 发布

Gogo-2020

最新推荐文章于 2024-09-24 08:39:28 发布

阅读量146

点赞数

分类专栏： leetcode 文章标签：动态规划 leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_44560620/article/details/116429132

版权

leetcode 专栏收录该内容

804 篇文章 3 订阅

订阅专栏

本文介绍了一种高效的算法，通过统计数组中每个元素的出现次数，结合动态规划求解如何在满足特定规则下最大化获得点数的问题。针对给定整数数组nums，演示了如何通过删除元素并根据相邻元素删除规则来计算最大收益。

摘要由CSDN通过智能技术生成

给你一个整数数组 nums ，你可以对它进行一些操作。

每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素。

开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。

示例 1：

输入：nums = [3,4,2]
输出：6
解释：
删除 4 获得 4 个点数，因此 3 也被删除。
之后，删除 2 获得 2 个点数。总共获得 6 个点数。

解题思路

代码分为两个部分

统计每个元素出现的次数（方便计算删除元素后的得分），并且找出最大值（为了缩小dp数组的长度）

	for _, num := range nums {
		cnt[num]++
		if num>max{
			max=num
		}
	}

状态转移
dp[i][0]代表当前元素是i，并且不删除该元素。因此前一个元素可以是被删除元素，也可以不是
dp[i][1]代表当前元素是i，需要删除该元素。因此前一个元素必须不为删除元素（因为如果前一个元素是删除元素，该元素已经被删除掉了），并且加上删除后的得分

	for i := 1; i <= max; i++ {

		dp[i][0]=MaxV(dp[i-1][0],dp[i-1][1])
		dp[i][1]=dp[i-1][0]+i*cnt[i]
	}

代码

func MaxV (a int,b int) int {
	if a>b{
		return a
	}else {
		return b
	}
}
  

func deleteAndEarn(nums []int) int {
	cnt:=make([]int,10008)
	max:=-1
	for _, num := range nums {
		cnt[num]++
		if num>max{
			max=num
		}
	}
	dp:=make([][2] int,max+1)
	for i := 1; i <= max; i++ {

		dp[i][0]=MaxV(dp[i-1][0],dp[i-1][1])
		dp[i][1]=dp[i-1][0]+i*cnt[i]
	}
    return MaxV(dp[max][0],dp[max][1])
}