题目说每人最少为1颗糖(即每个dp的初始化可以先设为1),且相邻两个分数高的糖要多。那么我们就可以发现当前位置的状态(糖果数量)和两边人的分数及糖果量有关。
但是,可能有人会问,相邻的状态可以由前一个已经得到的状态转移可以理解,但是右边一个人的状态我还没遍历到,怎么利用呢?
这时候,只需要进行两次顺序和逆序循环遍历就行了
举个例子
[1,0,2]
初次顺序遍历(只看当前是否比左边大,大就dp+1)
分糖:
1->1->2
进行一次逆序 (只比是否比右边大且糖果还比右边少的情况)
2->1->2
即是题目答案。
代码化:
long long dp[50000+500];
int candy(int* ratings, int ratingsSize){
int i, j;
for(i=0;i<ratingsSize;i++)
{
dp[i]=1;
}
for(i=1;i<ratingsSize;i++)
{
if(ratings[i]>ratings[i-1])
dp[i]=dp[i-1]+1;
}
for(i=ratingsSize-2;i>=0;i--)
{
if(ratings[i]>ratings[i+1]&&dp[i]<=dp[i+1])
dp[i]=dp[i+1]+1;
}
long long sum=0;
for(i=0;i<ratingsSize;i++)
sum+=dp[i];
return sum;
}