> 文章列表 > python3-算法刷题-贪心算法-更新中

python3-算法刷题-贪心算法-更新中

python3-算法刷题-贪心算法-更新中

605. 种花问题

https://leetcode.cn/problems/can-place-flowers

假设有一个很长的花坛,一部分地块种植了花,另一部分却没有。可是,花不能种植在相邻的地块上,它们会争夺水源,两者都会死去。

给你一个整数数组 flowerbed 表示花坛,由若干 0 和 1 组成,其中 0 表示没种植花,1 表示种植了花。另有一个数 n ,能否在不打破种植规则的情况下种入 n 朵花?能则返回 true ,不能则返回 false 。

示例 1:
输入:flowerbed = [1,0,0,0,1], n = 1
输出:true

示例 2:
输入:flowerbed = [1,0,0,0,1], n = 2
输出:false

提示:
1 <= flowerbed.length <= 2 * 10^4
flowerbed[i] 为 0 或 1
flowerbed 中不存在相邻的两朵花
0 <= n <= flowerbed.length

思路:
遍历数组。如果当前index是i,数值是1,那么下一位肯定不能种,只能考虑下下一个,所以直接跳去 i + 2;如果当前数值是0,说明左邻肯定不是1,因为如果是1的话i会被跳过,那么只考虑右邻是不是0或者到达边界就好了。此时如果满足二者之一,就可以在i位种树,那么n–,并且因为i位种了,相当于i位数值是1,所以直接跳去i + 2;如果两个条件都没有满足,说明不仅当前不能种,而且因为右邻是1也不能种,所以一次要跳1 + 2 = 3,因此跳去i + 3。如果最后n <= 0,说明所有的花都种进去了,自然返回True。

class Solution:def canPlaceFlowers(self, flowerbed: List[int], n: int) -> bool:i = 0size = len(flowerbed)while i < size:if n == 0: return Trueif flowerbed[i] == 1:i += 2else:if i + 1 < size and flowerbed[i + 1] == 0 or i + 1 == size:n -= 1i += 2else:i += 3return n <= 0

122. 买卖股票的最佳时机 II

https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-ii/

给你一个整数数组 prices ,其中 prices[i] 表示某支股票第 i 天的价格。
在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。
返回 你能获得的 最大 利润

示例 1:
输入:prices = [7,1,5,3,6,4]
输出:7
解释:在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。
随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6 - 3 = 3 。
总利润为 4 + 3 = 7 。

示例 2:
输入:prices = [1,2,3,4,5]
输出:4
解释:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4 。
总利润为 4 。

示例 3:
输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 交易无法获得正利润,所以不参与交易可以获得最大利润,最大利润为 0 。

提示:1 <= prices.length <= 3 * 10^4
0 <= prices[i] <= 10^4

思路:
如果明天比今天的价格高,就今天买明天卖,获取盈利;否则不买不卖

class Solution:def maxProfit(self, prices: List[int]) -> int:if len(prices) == 1: return 0ans = 0for i in range(len(prices) - 1):ans += max(prices[i + 1] - prices[i], 0)return ans