LeetCode 575. 分糖果

Posted 2021-05-23 数据结构和算法

想看更多算法题，可以扫描上方二维码关注我微信公众号“数据结构和算法”，截止到目前我已经在公众号中更新了500多道算法题，其中部分已经整理成了pdf文档，截止到目前总共有1000多页（并且还会不断的增加），可以在公众号中回复关键字“pdf”即可下载。

public int candy(int[] ratings) {
    int length = ratings.length;
    //记录的是从左往右循环的结果
    int[] left = new int[length];
    //记录的是从右往左循环的结果
    int[] right = new int[length];
    //因为每个孩子至少有一个糖果，默认都给他们一个
    Arrays.fill(left, 1);
    Arrays.fill(right, 1);
    //统计最少的总共糖果数量
    int total = 0;
    //先从左往右遍历，如果当前孩子的得分比左边的高，
    //那么当前孩子的糖果要比左边孩子的多一个
    for (int i = 1; i < length; i++) {
        if (ratings[i] > ratings[i - 1])
            left[i] = left[i - 1] + 1;
    }
    //然后再从右往左遍历，如果当前孩子的得分比右边的高，
    //那么当前孩子的糖果要比右边边孩子的多一个
    for (int i = length - 1; i > 0; i--) {
        if (ratings[i - 1] > ratings[i])
            right[i - 1] = right[i] + 1;
    }
    //要满足左右两边的条件，那么当前孩子的糖果就要取
    //从左往右和从右往左的最大值。
    for (int i = 0; i < length; i++) {
        //累计每个孩子的糖果
        total += Math.max(left[i], right[i]);
    }
    return total;
}

其实还可以优化一下，在从右往左遍历的时候就可以统计total的值了，不需要再使用第3个for循环，代码如下

public int candy(int[] ratings) {
    int length = ratings.length;
    int[] count = new int[length];
    //默认给每个孩子一个糖果
    Arrays.fill(count, 1);
    //先从左往右遍历
    for (int i = 1; i < length; i++) {
        if (ratings[i] > ratings[i - 1])
            count[i] = count[i - 1] + 1;
    }
    //因为下面的for循环中，total没有统计最后一个孩子的糖果，所以这里total
    //的默认值就是最后那个孩子的糖果数量
    int total = count[length - 1];
    //从右往左遍历，然后顺便累加total的值
    for (int i = length - 1; i > 0; i--) {
        if (ratings[i - 1] > ratings[i])
            count[i - 1] = Math.max(count[i - 1], count[i] + 1);
        total += count[i - 1];
    }
    return total;
}

public int candy(int[] ratings) {
    int length = ratings.length;
    if (length == 1)
        return 1;
    //记录访问到哪个孩子了
    int index = 0;
    //记录总共的糖果
    int total = 0;
    while (index < length - 1) {
        int left = 0;
        //统计递增的长度
        while (index < length - 1 && ratings[index + 1] > ratings[index]) {
            index++;
            left++;
        }
        int right = 0;
        //统计递减的长度
        while (index < length - 1 && ratings[index + 1] < ratings[index]) {
            index++;
            right++;
        }
        //记录顶点的值，也就是左右两边最大的值加1
        int peekCount = Math.max(left, right) + 1;
        //注意这里如果total不等于0要减1，是因为把数组拆分成子数组的时候，低谷的那
        //个会被拆到两个数组中，如果total不等于0，说明之前已经统计过，而下面会再次
        //统计，所以要提前减去。
        if (total != 0)
            total--;
        //当前这个子数组的糖果数量就是前面递增的加上后面递减的然后在加上顶点的。
        total += (1 + left) * left / 2 + (1 + right) * right / 2 + peekCount;
        //如果当前孩子的得分和前一个一样，我们让他降为1
        while (index < length - 1 && ratings[index + 1] == ratings[index]) {
            index++;
            total++;
        }
    }
    return total;
}