题目
思路
数组是无序的,删除元素不方便,所以先变得有序,怎么变得有序? 创建新的数组arr,数组的大小是原数组中的最大数+1,原数组的元素对应arr下标,然后遍历原数组,在对应的位置加加,目的是统计下标值在原数组出现的总值。
没听懂没关系,举个例子就容易理解了,假设原数组{1,2,2,4,4,4,7,7,8},建立arr[9]的临时数组,初始化arr数组为0,遍历完原数组后arr中的值为{0,1,4,0,12,0,0,14,8},其中arr[2]表示原数组中的2相加后为4。
变成有序后,有没有发现问题就变成了:不能取相邻点数值求最大点数。类似于“按摩师”问题。具体步骤在下面链接里,在这里不再赘述。
代码
class Solution {
public:
int rob(int*arr,int len)
{
int f[len];
int g[len];
f[0]=arr[0];
g[0]=0;
for(int i=1;i<len;i++)
{
f[i]= g[i-1]+arr[i];
g[i]=max(g[i-1],f[i-1]);
}
return max(f[len-1],g[len-1]);
}
int deleteAndEarn(vector<int>& nums) {
int max=nums[0];
int len=nums.size();
int i;
//找出最大值
for(i=0;i<len;i++)
{
if(max<nums[i])
max=nums[i];
}
int arr[max+1];
//初始化arr
for(i=0;i<max+1;i++)
{
arr[i]=0;
}
for(i=0;i<len;i++)
{
arr[nums[i]] +=nums[i];
}
//对arr中的数进行“打家劫舍”/"按摩师问题"
int ret = rob(arr,max+1);
return ret;
}
};