分发糖果

/*
 * 思路：
 * 1. 首先初始化每个孩子至少得到1个糖果。
 * 2. 从左到右遍历，如果右边的孩子评分比左边的高，右边孩子的糖果数为左边孩子糖果数加1。
 * 3. 从右到左遍历，如果左边的孩子评分比右边的高，左边孩子的糖果数为右边孩子糖果数加1。
 * 4. 使用Java Stream返回所有孩子糖果数的总和。
 */
 
import java.util.*;
import java.util.stream.*;
 
public class Solution {
    public int candy(int[] ratings) {
        int n = ratings.length;
        int[] candies = new int[n];
        Arrays.fill(candies, 1);
 
        // 从左到右遍历
        IntStream.range(1, n).forEach(i -> {
            if (ratings[i] > ratings[i - 1]) {
                candies[i] = candies[i - 1] + 1;
            }
        });
 
        // 从右到左遍历
        IntStream.iterate(n - 2, i -> i >= 0, i -> i - 1).forEach(i -> {
            if (ratings[i] > ratings[i + 1]) {
                candies[i] = Math.max(candies[i], candies[i + 1] + 1);
            }
        });
 
        // 计算总糖果数
        return Arrays.stream(candies).sum();
    }
}