(LeetCode 面试经典 150 题) 135. 分发糖果（贪心）

岁忧

于 2025-06-30 18:16:51 发布

阅读量236

点赞数 8

CC 4.0 BY-SA版权

分类专栏： LeetCode 面试经典 150 题 LeetCode C++、JAVA、Go版本文章标签： leetcode 面试算法 c++ java go

本文链接：https://blog.csdn.net/weixin_46028214/article/details/149028845

LeetCode 同时被 3 个专栏收录

809 篇文章

订阅专栏

C++、JAVA、Go版本

98 篇文章

订阅专栏

LeetCode 面试经典 150 题

16 篇文章

订阅专栏

题目：135. 分发糖果

在这里插入图片描述
思路：贪心，时间复杂度0(n)。
先给所有人一颗糖果，然后从左到右遍历，ratings[i]>ratings[i-1]时，分配的糖果比前一个多一个：v[i]=v[i-1]+1。
在从右到左遍历，当ratings[i]>ratings[i+1]时，维护v[i]分配的糖果比后一个v[i+1]多1个,：v[i]=max(v[i],v[i+1]+1)。

C++版本：

class Solution {
public:
    int candy(vector<int>& ratings) {
        int n=ratings.size();
        vector<int> v(n,1);
        for(int i=1;i<n;i++){
            if(ratings[i]>ratings[i-1]){
                v[i]=v[i-1]+1;
            }
        }
        for(int i=n-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]){
                v[i]=max(v[i],v[i+1]+1);
            }
        }
        int sum=0;
        for(auto x:v){
            sum+=x;
        }
        return sum;
    }
};

JAVA版本：

class Solution {
    public int candy(int[] ratings) {
        int n=ratings.length;
        int[] v=new int[n];
        Arrays.fill(v,1);
        for(int i=1;i<n;i++){
            if(ratings[i]>ratings[i-1]){
                v[i]=v[i-1]+1;
            }
        }
        for(int i=n-2;i>=0;i--){
            if(ratings[i]>ratings[i+1]){
                v[i]=Math.max(v[i],v[i+1]+1);
            }
        }
        int sum=0;
        for(var x:v){
            sum+=x;
        }
        return sum;
    }
}

Go版本：

func candy(ratings []int) int {
    n:=len(ratings)
    v:=make([]int,n)
    for i:=1;i<n;i++ {
        if ratings[i]>ratings[i-1] {
            v[i]=v[i-1]+1
        }
    }
    for i:=n-2;i>=0;i-- {
        if ratings[i]>ratings[i+1] {
            v[i]=max(v[i],v[i+1]+1)
        }
    }
    sum:=n
    for _,x:=range v {
        sum+=x
    }
    return sum
}