740. Delete and Earn

来源:互联网 发布:无人机飞控调参软件 编辑:程序博客网 时间:2024/05/18 06:48

这也是之前没有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];    }};
原创粉丝点击