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;
}

};