Leetcode之Candy分糖果问题

题目：

题目大意：几个小孩站一排，每个小孩有个等级值，现在给小孩发糖，发的时候要遵守2个规则：
（1）每个小孩至少一颗糖
（2）两个相邻的小孩中，等级大的小孩一定比等级小的小孩糖多，求发糖的数目的最小值。（等级相同的可以有多有少）

解法1：

刚刚开始所有小孩的糖果数都为1，从左往右，arr[i]>arr[i-1]则当前i的小孩糖果数加1，再从右往左，arr[i]<arr[i+1]则当前i的小孩糖果数加1，时间复杂度O（n），空间复杂度O（n）。

代码：

public static int candy1(int[] arr){
        int len = arr.length;
        int[] res = new int[len];
        for(int i = 0; i < len; i++){
            res[i] = 1;
        }
        for(int i = 1; i < len; i++){
            if(arr[i] > arr[i-1]){
                res[i] = res[i-1]+1;
            }
        }
        for(int i = len-2; i >= 0; i--){
            if(arr[i] > arr[i+1]){
                res[i] = res[i+1]+1;
            }
        }
        int sum = 0;
        for(int i = 0; i < len; i++){
            sum += res[i];
        }
        return sum;
    }

解法2：

刚刚开始给第一个孩子1个糖果，后面的情况可能是：

（1）arr[i]>arr[i-1]，则当前i发放糖果数为pre+1；

（2）arr[i]=arr[i-1]，则当前i发放糖果数为1；

（3）arr[i]<arr[i-1]，则当前i发放糖果数未知，我们会统计下坡的个数，然后在（1）情况中进行处理。

时间复杂度O（n），空间复杂度O（1）。

代码：

public static int candy2(int[] arr){
        //pre代表遍历到i时前一个位置i-1分到的糖果数
        //count是每次下坡的元素个数，total总糖果数
        int pre = 1, count = 0, total = 1;
        for(int i = 1; i < arr.length; i++){
            if(arr[i] >= arr[i-1]){
                //在开始上坡时处理前面的降坡
                if(count > 0){
                    total += count*(count+1)/2; //从右往左开始，1.2.3.....count
                    //与一个问题，前一个pre(前一个上坡最高点)小于count，则这个pre应该修改为count+1
                    if(count > pre){
                        total += count+1-pre;
                    }
                    count = 0; //count清零
                    pre = 1;//之前的pre为前一个升坡最高点，需要更新为1，可以理解为谷底，下一个上波的开始
                }
                pre = arr[i] == arr[i-1] ? 1 : pre+1; //相等，此时i处小孩分糖果为1，否则前者+1
                total += pre;
            }else {
                count++; //下坡数统计
            }
        }
        //如果遍历到最后一个数时为下坡，count需要处理
        if(count > 0){
            total += count*(count+1)/2;
            if(count > pre){
                total += count+1-pre;
            }
        }
        return total;
    }