代码随想录算法训练营第三十一天|455. 分发饼干,376. 摆动序列,53. 最大子数组和
贪心算法,做了无数次依然不知道要干什么的算法
455. 分发饼干
题目链接:分发饼干
先满足胃口大的孩子
class Solution:
def findContentChildren(self, g: List[int], s: List[int]) -> int:
g.sort()
s.sort()
start = len(s)-1
count = 0
for i in range((len(g)-1),-1,-1):
if start >= 0 and s[start] >= g[i]:
start -= 1
count += 1
return count
在先满足胃口大的孩子的情况下,是不可遍历饼干的,因为会出现没有一块饼干可以满足胃口最大的小朋友,遍历也结束了,后面的都没做到。
先满足胃口小的孩子
class Solution(object):
def findContentChildren(self, g, s):
"""
:type g: List[int]
:type s: List[int]
:rtype: int
"""
g.sort()
s.sort()
res = 0
for i in range(len(s)):
if res <= (len(g)-1) and s[i] >= g[res]:
res += 1
return res
这里可以遍历饼干,因为如果没有一块饼干可以满足胃口最小的小朋友,那就是0了。
376. 摆动序列
题目链接:摆动序列
视频解析
这里的想法是,如果有单调递增或者单调递减的中间序列(单调坡),我们考虑把坡的两头留下,中间全删掉。
这是一种贪心算法。
我一开始的想法是三个指针跳着走,但是后来跳少了,这个题是输出列表个数,完全不需要知道到底是谁在最后的list里,意思就是单调坡的时候,选出其中两个就行,是谁不重要。
最重要的是怎么处理坡和平坡的留下,这里善用等于。
- nums = [1,2,2,1],这是个平坡,我们要想好留第一个2还是第二个2,如果没有放任何diff == 0的条件的话,出来的结果会只有1一个单摆序列。所以这里,如果我们允许prediff = 0,那么就是要后一个2,如果允许curdiff = 0,那么就是要第二个2。
- 由于无论如何,我们摆动序列都会有一个,如果序列是两个不相等的元素组成的那就有两个,考虑到上面平坡的情况,我们可以初始化prediff = 0,意思是在nums[0]前面装一个和他相等的元素,然后允许prediff = 0,那么就留下后一个(留下nums[0]本体),如果curdiff还有摆动,那么就是两个,如果curdiff不摆动,那就还是一个。
- 接下来是怎么更新prediff的问题,nums = [1,2,2,3],这是个单调加平坡,这种情况下单摆序列是一个,如果我们每次都更新prediff,完全跟着curdiff走,那么就会算出2个,因为他们不知道坡前面那个差是增的还是减的,所以我们每次开始摆动是才更新prediff,这样就可以和curdiff对上。
class Solution:
def wiggleMaxLength(self, nums: List[int]) -> int:
if len(nums) == 1: return 1
result = 1
prediff = 0
curdiff = 0
for i in range(1,len(nums)):
curdiff = nums[i] - nums[i-1]
if (prediff <= 0 and curdiff > 0) or (prediff >= 0 and curdiff < 0):
result += 1
prediff = curdiff
return result
其实几种情况都想到了,但是实现真的很难,也没想到代码这么短就可以。
53. 最大子数组和
题目链接:最大子数组和
这个题的贪心之处在于,只要现在的sum<0,那就直接丢掉重新开始。
本来想用滑动窗口做,但是怎么丢弃完全没有思路。
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
start = 0
s = 0
max_s = -inf
while start < len(nums):
s += nums[start]
if s > max_s:
max_s = s
if s < 0:
s = 0
start += 1
else:
start += 1
return max_s