题意:
一群孩子做游戏,现在请你根据游戏得分来发糖果,要求如下:
1. 每个孩子不管得分多少,起码分到一个糖果。
2. 任意两个相邻的孩子之间,得分较多的孩子必须拿多一些糖果。(若相同则无此限制)
给定一个数组 arrarr 代表得分数组,请返回最少需要多少糖果。
要求: 时间复杂度为 O(n)O(n) 空间复杂度为 O(n)O(n)
数据范围: 1 \le n \le 1000001≤n≤100000 ,1 \le a_i \le 10001≤ai≤1000
示例:
输入:
[1,1,2]返回值:
4说明:
最优分配方案为1,1,2
题解:首先给有人分1个糖果,其次正向遍历,如果分数大于前一个人,则糖果数为前一个人的糖果数+1,但无法保证当前拥有的糖果数和后一个人的糖果数满足要求,所以逆向遍历,如果分数大于后一个人且糖果数小于等于后一个人的糖果数,则糖果数为后一人的糖果数+1
import java.util.*;
public class Solution {
/**
* pick candy
* @param arr int整型一维数组 the array
* @return int整型
*/
public int candy (int[] arr) {
// write code here
int len=arr.length,m=0;
int [] s=new int [len];
for(int i=0;i<len;++i)
s[i]=1;
for(int i=1;i<len;++i)
{
if(arr[i]>arr[i-1])
{
s[i]=s[i-1]+1;
}
}
for(int i=len-2;i>=0;i--)
{
if(arr[i]>arr[i+1]&&s[i]<=s[i+1])
{
s[i]=s[i+1]+1;
}
}
for(int i=0;i<len;++i)
m+=s[i];
return m;
}
}