面试题：分发糖果——JAVA

问题描述

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

1. 每个孩子至少分配到 1 个糖果。
2. 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的最少糖果数目。

解题思路

这是一个经典的贪心算法问题。采用两次遍历的方法：

1. 从左到右遍历一次，确保右边评分高的孩子获得更多糖果。
2. 从右到左遍历一次，确保左边评分高的孩子获得更多糖果。

最终，每个孩子得到的糖果数目就满足题目的条件。

代码实现（Java）

class Solution {
    public int candy(int[] ratings) {
        int n = ratings.length;
        
        // 初始化糖果数组，每个孩子至少一个糖果
        int[] candies = new int[n];
        Arrays.fill(candies, 1);
        
        // 从左到右遍历，确保右边评分高的孩子获得更多糖果
        for (int i = 1; i < n; i++) {
            if (ratings[i] > ratings[i - 1]) {
                candies[i] = candies[i - 1] + 1;
            }
        }
        
        // 从右到左遍历，确保左边评分高的孩子获得更多糖果
        for (int i = n - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1]) {
                candies[i] = Math.max(candies[i], candies[i + 1] + 1);
            }
        }
        
        // 计算总糖果数
        int totalCandies = 0;
        for (int candy : candies) {
            totalCandies += candy;
        }
        
        return totalCandies;
    }
}

 测试

public static void main(String[] args) {
        Solution solu = new Solution();
        
        // 示例
        int[] ratings1 = {1, 0, 2};
        int[] ratings2 = {1, 2, 2};

        int result1 = solu.candy(ratings1);
        int result2 = solu.candy(ratings2);

        System.out.println(result1);  // 输出：5
        System.out.println(result2);  // 输出：4
    }