LeetCode 135. Candy

这个题目难度很小,也是一道贪心的题目。怎么贪呢?就按照题目要求来,可是遇到这个问题怎么办:后一个比前一个小,但是后一个却是一个大于它之后数的。这个会导致选择困难,你没法选出一个优解。我的解决办法是,排序,然后对排序后的序列的索引一个一个找周围的贪心解。这样就可以很好的,解决这个问题。其实还有一种更简单的办法,就是前向,找前一个比后一个大的计算好数组后,在后向找后一个比前一个大的。就可以了。我的代码如下:

class Solution {
public:
    struct cmp{
      bool operator()(const pair<int,int> &a,const pair<int,int> &b){
        return a.second <= b.second;
    } 
    };
    int candy(vector<int>& ratings) {
        int len = ratings.size();
        if(len==1)return 1;
        int *dp = new int[len];    
        vector<pair<int,int>> p;
        map<int,int>s;
        for(int i=0;i<len;i++)
        {   
            s.insert(pair<int,int>(i,ratings[i]));
        }
        map<int,int>::iterator ptr1;
        for(ptr1 = s.begin();ptr1!=s.end();++ptr1)
        p.push_back(make_pair(ptr1->first,ptr1->second));
        sort(p.begin(),p.end(),cmp());
        for(int i = 0;i<len;i++)
        dp[i]=1;
        vector<pair<int,int>>::iterator ptr;
        int res = 0;
        for(ptr=p.begin();ptr!=p.end();++ptr)
        {   int cur = ptr->first;
            //cout<<ratings[cur]<<endl;
            if(((ptr->first)-1)>=0&&((ptr->first)+1)<len)
            {
                int pre = (ptr->first)-1,next = (ptr->first)+1;
              //  cout<<"pre: "<<ratings[pre]<<" "<<"cur: "<<ratings[cur]<<"next: "<<ratings[next]<<endl;
                if(ratings[cur]>ratings[pre]&&ratings[cur]<=ratings[next])
                {
                    dp[cur] = dp[pre]+1;
                }
                else if(ratings[cur]>ratings[next]&&ratings[cur]<=ratings[pre])
                {
                    dp[cur] = dp[next]+1;
                }
                else if(ratings[cur]>ratings[pre]&&ratings[cur]>ratings[next])
                {
                    
                    dp[cur] = max(dp[next],dp[pre])+1;
                }
            }
            else if(((ptr->first)-1)<0)
            {    int next = (ptr->first)+1;
                 if(ratings[next]<ratings[cur])dp[cur]=dp[next]+1;
            }
            else if(((ptr->first)+1)>=len)
            {   int pre = (ptr->first)-1;
                if(ratings[pre]<ratings[cur])dp[cur]=dp[pre]+1;
            }
            res+=dp[cur];
        }
        /*cout<<endl;
        for(int i = 0;i<len;i++)
        cout<<dp[i]<<endl;*/
        return res;
    }

};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值