leetcode

来源:互联网 发布:单片机恒温控制系统 编辑:程序博客网 时间:2024/06/16 06:57

算法系列博客之Greedy

Greedy 贪心算法是一种非常优美的算法,不过贪心算法本身的可行性很多时候会受到一些局限。但是一旦能够找到一种可行的贪心策略,问题的解决将会变得非常高效,因为通常情况下,贪心算法的复杂度是O(n)

本篇博客将运用这种思想来解决leetcode上135号问题


问题描述:

There are N children standing in a line. Each child is assigned a rating value.

You are giving candies to these children subjected to the following requirements:

Each child must have at least one candy.
Children with a higher rating get more candies than their neighbors.
What is the minimum candies you must give?

从贪心的角度出发,目的是给每个孩子尽量少的糖,我们很容易想到这样一种策略:
给左侧第一个孩子一颗糖,后面的孩子都和他的前一个作比较
      ·   如果rating值比其左侧孩子大,则让他得到的糖的个数比其左侧孩子的糖的个数大1
      ·   否则(即rating值小于或者等于其左侧孩子),只给他一颗糖
但是仔细研究,我们发现,这样存在一个问题:
       题中的要求是如果一个孩子比他旁边的孩子rating值大,就应该拥有比其旁边多的糖,这里的旁边是指两侧;
       然而,目前的策略有可能会造成某个孩子比右侧的孩子rating值大,但是他们都只拥有1颗糖
乍一看好像这种贪心策略在此不可行,但是细心一点就会发现,它至少保证了前进方向上满足规则
那么我们将前进方向反过来再来一次不就可以保证两个方向(也就是每个孩子两侧)都满足规则了吗?
不过反过来再来一次时初始状态就不再一样,所以策略需要做一点点微调:
      ·   如果rating值比右侧孩子大并且他拥有的糖的个数小于等于右侧孩子,则让他得到的糖的个数比前一个孩子大1
      ·   否则(即rating值小于或者等于前一个孩子),不更改他所拥有的糖的个数
其python 代码实现如下:

class Solution(object):    def candy(self, ratings):        n = len(ratings)        candys = [1] * n        for i in range(1, n):            if (ratings[i] > ratings[i-1]):                candys[i] = candys[i-1] + 1        res = candys[n-1]        for i in range(n-2, -1, -1):            if (ratings[i] > ratings[i+1] and candys[i] <= candys[i+1]):                candys[i] = candys[i+1] + 1            res += candys[i]        return res

两次循环使得每个孩子的左右两侧都满足性质,因而这种贪心策略是可行的,算法的正确性可以很容易得到证明

时间复杂度分析,两次并行线性循环,循环内部都是常数运算操作,因而O(n)
空间复杂度分析,需要开辟一个和rating一样大小的数组,因而O(n)
此刻我们可以说,这种时间和空间上都能达到线性复杂度的贪心算法是令人满意的

0 0
原创粉丝点击