这也是之前没有ac的代码,超时了,因此采用这种方式,先统计每种数字删除后的value,然后利用dp【i】算法,计算[i, end],这个范围内的,和的最大值。首先需要明确,如果采用二维dp,很容易超时。
class Solution {
public:
int deleteAndEarn(vector<int>& nums) {
map<int,int> count;
for(int i=0;i<nums.size();i++)
{
if(count.find(nums[i])!=count.end())
count[nums[i]]+=1;
else
count[nums[i]]=1;
}
vector<int> numbers;
vector<int> values;
vector<int> dp;
for(map<int,int>::iterator it=count.begin();it!=count.end();++it)
{
numbers.push_back(it->first);
values.push_back((it->first)*(it->second));
dp.push_back(0);
}
dp.push_back(0);
for(int i=numbers.size()-1;i>=0;i--)
{
int sum=0;
sum=values[i];
if(i+1<numbers.size())
{
if(numbers[i+1]==numbers[i]+1)
sum+=dp[i+2];
else
sum+=dp[i+1];
dp[i] = sum>dp[i+1]?sum:dp[i+1];
}
else
dp[i]=values[i];
//cout<<i<<" "<<dp[i]<<endl;
}
return dp[0];
}
};