LeetCode 135. Candy

来源:互联网 发布:传奇霸业轮回魔道数据 编辑:程序博客网 时间:2024/06/05 14:57

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

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
原创粉丝点击