Java 贪心算法 大厂面试题 230729

112 阅读1分钟

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;

    }