本文主要是介绍【小白友好】LeetCode 删除并获得点数,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
基础题
打家劫舍https://leetcode.cn/problems/house-robber/
小白解法
- 删除nums[i]就会使得所有
nums[i]-1
和nums[i]+1
的值都消失,手写了几个,发现找来找去不方便,还不如先排个序,然后这样nums[i]-1
和nums[i]
和nums[i]+1
就能靠在一起了,这样删的时候方便找。 - 只要获得一次nums[i]的点数那么肯定是要一起把所有的nums[i]都要带上的,也就是获得
nums[i]*nums[i]次数
个点数
嗯,发现[1,1,1,2,2,3,4,5]
的时候怎么这么熟悉,选1就不能选2,选2就不能选3,这不是相邻的2个数不能一起选?这不是打家劫舍吗?
但是也有所区别,当相邻的2个值相差不是1时,可以直接获得,不需要“打家劫舍”。我们此时说的“相邻”是在[1,2,3,4]
这个新构造的数组上的相邻,而不是原数组[1,1,1,2,2,3,4,5]
未知的相邻。 也就是把每个相同元素的数字看成1个数,其对应的能获得点数是nums[i]*nums[i]次数
。
那么要怎么新构造的数组?可以使用C++风格的直接[0 for i in range(max(nums))]
构造一个可能超长的数组,然后把对应位置的元素填上,解起来就跟打家劫舍的写法一模一样了;也可以使用一个字典num2value
,统计key能获得的value,然后在判断的时候加入是否abs(nums[i]-nums[i-1])==1
。
我个人还是倾向于字典的写法:
class Solution:def deleteAndEarn(self, nums: List[int]) -> int:# 统计每个数字对应能获得的点数from collections import defaultdictnum2value=defaultdict(int)for n in nums:num2value[n]+=n# 构造新数组sorted_nums sorted_nums=sorted(set(nums))if len(sorted_nums)==1:return num2value[nums[0]]if len(sorted_nums)==2:if abs(sorted_nums[0]-sorted_nums[1])==1:return max(num2value[sorted_nums[0]],num2value[sorted_nums[1]])else:return num2value[sorted_nums[0]]+num2value[sorted_nums[1]]# 上述特殊情况# 下面是初始化+转移方程dp=[0 for _ in range(len(sorted_nums))]dp[0]=num2value[sorted_nums[0]]if abs(sorted_nums[0]-sorted_nums[1])==1:# 如果是相邻元素,那么就是打家劫舍式的更新dpdp[1]= max(num2value[sorted_nums[0]],num2value[sorted_nums[1]])else:# 若不是,可以直接+,不受影响dp[1]= num2value[sorted_nums[0]]+num2value[sorted_nums[1]]for i in range(2,len(sorted_nums)):if abs(sorted_nums[i]-sorted_nums[i-1])!=1:dp[i]=dp[i-1]+num2value[sorted_nums[i]]else:select_now=num2value[sorted_nums[i]]+dp[i-2]unselect_now=dp[i-1]dp[i]=max(select_now,unselect_now)# print(dp)return dp[-1]
在写完下方的一般情况后,仍然不要忘了特殊情况,下方的下标是有i-1和i-2的,这两种需要单独去返回。
不得不说小白写法写的真的很繁琐,不优美,但是便于理解。
这篇关于【小白友好】LeetCode 删除并获得点数的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!