传送门
题目:给定一个整数数组 nums ,你可以对它进行一些操作。每次操作中,选择任意一个 nums[i] ,删除它并获得 nums[i] 的点数。之后,你必须删除每个等于 nums[i] - 1 或 nums[i] + 1 的元素。
开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。 输入: nums = [2, 2, 3, 3, 3, 4] 输出: 9 解释: 删除 3 来获得 3 个点数,接着要删除两个 2 和 4 。 之后,再次删除 3 获得 3 个点数,再次删除 3 获得 3 个点数。 总共获得 9 个点数。 注意: 1.nums的长度最大为20000。 2.每个整数nums[i]的大小都在[1, 10000]范围内。
不要在意这里要删除i-1和i+1,虽然这里是删除两个相邻的,但是换一个角度:
删除相邻的两个数,不就相当于选这个数的时候,跟他相邻的两个数都不选吗!
这不就是打家劫舍的动态规划过程吗。
你可能会想这道题要删除的两个数可能在nums里不和i相邻,那怎么办?
好办。用hash表(大小是nums里的最大值,这里题目说nums[i]<=1000,就可以直接设成1000)存: hash[i] 就是 数字i在nums中出现的次数。 这样,在更新dp的时候,i从2遍历到1000(0不用管,1特判) i就代表要取得数值(相当于i=nums[j])
i*count[i]就是这个数一共可以贡献的和。
所以,最后for里面不是nums得长度,是count得长度(1001)
public int deleteAndEarn(int[] nums) { if (nums.length == 0) return 0; int[] count = new int[10001]; //统计每个num出现次数 int[] dp = new int[10001]; for (int num : nums) count[num]++; dp[1] = 1 * count[1]; for (int i = 2; i < count.length; ++i) { dp[i] = Math.max(dp[i - 1], dp[i - 2] + i * count[i]); } return dp[dp.length - 1]; }