打家劫舍算法

打家劫舍算法

题目描述

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

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

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

解题思路

1. 首先将每个相同的数字整体求和
2. 对于求和出的数组来说，就是一个打家劫舍问题

解题流程

对于一个数组nums，对于前两个元素进行判断，first = nums[0]、second = Math.max(nums[0],nums[1]);
上图是nums[0] < nums[1]
下图是nums[0] > nums[1]

然后从第三个位置(i = 2)开始判断，如果first + nums[2] > second，则更新second，否则不进行更新。然后将原来second的值赋给first

这样就能保住first永远是小于等于second的，而second的值是i前的最大值

代码

class Solution {
    public int deleteAndEarn(int[] nums) {
        int maxint = 0;
        for (int num : nums) {
            maxint = Math.max(num,maxint);
        }
        int[] sum = new int[maxint+1];
        for (int num : nums) {
            sum[num] += num;
        }
        return rob(sum);
        
    }
    public int rob(int[] nums){
        int size = nums.length;
        int first = nums[0];
        int second = Math.max(nums[0],nums[1]);
        for (int i = 2; i <size ; i++) {
            int temp = second;
            second = Math.max(first + nums[i],second);
            first = temp;
        }
        return second;
    }
}