JAVA 贪心算法 大厂面试题 230729-czl -leetcode

135. 分发糖果

贪心算法：

建立规则：设学生A B 左右相邻，A在B左边

​ 左规则：当ratings[i] > ratings[i-1] 时，B的糖比A的多

​ 右规则：当ratings[i] > ratings[i+1] 时，A的糖比B的多

相邻的学生中，评分高的学生必须获得更多的糖果 **等价于** 所有学生满足左规则且满足右规则。

重点：每次只考虑一个规则，不能同时考虑

• 算法流程：

1. 先从左至右遍历学生成绩 ratings

   • 第一个学生糖为1

   • 如果 ratings[i] > ratings[i-1] ，第 i个人糖果就比 第i-1 个人多一个,res[i；否则就只有一个；

     res[i] = res[i] = (ratings[i]>ratings[i-1]) ? res[i-1]+1 : 1 ;

2. 再从左到右遍历

   • 若ratings[i] > ratings[i+1] ，res[i] = max ( 第i个人的糖果数，第i+1 个人糖果数+1）

代码：

    public int candy(int[] ratings) {
        int size = ratings.length;
        if(size ==1){
            return size;
        }
        int[] res = new int[size];
        res[0]=1;

        for(int i = 1;i<size;i++){
            res[i] = (ratings[i]>ratings[i-1] ? res[i-1]+1:1   );
        }
        for(int i = size-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]){
                res[i] = Math.max(res[i+1]+1,res[i]);
            }
        }
      

        int sum = 0;
        for(int i = 0;i<size;i++){
            sum +=res[i];
        }

        return sum;

    }