这题可以化成与打家劫舍相似的格式来求解
为什么这么说呢?
回忆打家劫舍,当小偷偷了当前房屋之后,就不再考虑上一间房屋和下一间房屋。
本题中,当取了某个数值num则不再考虑num-1和num+1。
但是不同点在于
打家劫舍的遍历是顺序的,从第一间房屋到最后一间房屋,但是本题中对数组的遍历是无序的,对于例子
[3,4,2]
我可以先获得4,之后消除3
也可以先获得3,之后消除4和2
并且,本题中数组的数值可能重复,可能出现如下情况:
[3,3,3,3,2,2,2,2]
为了简化计算,先考虑数组中重复的情况,就上面的情况而言,第一次取3,则会消除所有的2,之后再取3,则不会对当前数组产生任何影响,因为在第一次取出时就已经被消除了,所以可以利用桶排序,记录出现的次数,这样一来,简化了取?和?删除?这两个操作。
并且,如果对比桶排序后的数组和打家劫舍的数组,会发现它们的意义是惊人的相似,
对于当前房屋会有偷与不偷两种选项,同样的,对于当前的数,会有取与不取的情况,
偷当前房屋,则不考虑上一个房屋和下一个房屋,同样的,对于取当前的数,则不考虑上一个数字和下一个数字
所以,我们可以用桶排序后的数组用动态规划,代码与打家劫舍相似
class Solution:
def deleteAndEarn(self, nums):
record=[0]*(max(nums)+1)
for index,num in enumerate(nums):
record[num]=record[num]+1
tmp={0:0,1:record[1]}
i=1
for i in range(2,len(record)):
tmp[i] = max(tmp[i-2]+record[i]*i,tmp[i-1])
return tmp[i]
|