170. Leetcode 135. 分发糖果 (贪心算法-两个维度权衡题目)
這道題目一定是要確定一邊之后,再確定另一邊,例如比較每一個(gè)孩子的左邊,然后再比較右 邊,如果兩邊一起考慮一定會(huì)顧此失彼。
先確定右邊評(píng)分大于左邊的情況(也就是從前向后遍歷)
此時(shí)局部最優(yōu):只要右邊評(píng)分比左邊大,右邊的孩子就多一個(gè)糖果,全局最優(yōu):相鄰的孩子中, 評(píng)分高的右孩子獲得比左邊孩子更多的糖果
局部最優(yōu)可以推出全局最優(yōu)。
如果ratings[i] > ratings[i - 1] 那么[i]的糖 一定要比[i - 1]的糖多一個(gè),所以貪心: candyVec[i] = candyVec[i - 1] + 1
再確定左孩子大于右孩子的情況(從后向前遍歷)
因?yàn)槿绻麖那跋蚝蟊闅v,根據(jù) ratings[i + 1] 來(lái)確定 ratings[i] 對(duì)應(yīng)的糖果,那么每次都 不能利用上前一次的比較結(jié)果了。
所以確定左孩子大于右孩子的情況一定要從后向前遍歷!
如果 ratings[i] > ratings[i + 1],此時(shí)candyVec[i](第i個(gè)小孩的糖果數(shù)量)就有兩個(gè) 選擇了,一個(gè)是candyVec[i + 1] + 1(從右邊這個(gè)加1得到的糖果數(shù)量),一個(gè)是 candyVec[i](之前比較右孩子大于左孩子得到的糖果數(shù)量)。
那么又要貪心了,局部最優(yōu):取candyVec[i + 1] + 1 和 candyVec[i] 最大的糖果數(shù)量, 保證第i個(gè)小孩的糖果數(shù)量即大于左邊的也大于右邊的。全局最優(yōu):相鄰的孩子中,評(píng)分高的孩 子獲得更多的糖果。
局部最優(yōu)可以推出全局最優(yōu)。
class Solution:def candy(self, ratings: List[int]) -> int:candy_res = [1] * len(ratings)for i in range(1, len(ratings)):if ratings[i] > ratings[i-1]:candy_res[i] = candy_res[i-1] + 1for j in range(len(ratings)-2,-1,-1):if ratings[j] > ratings[j+1]:candy_res[j] = max(candy_res[j],candy_res[j+1] + 1)return sum(candy_res)?
總結(jié)
以上是生活随笔為你收集整理的170. Leetcode 135. 分发糖果 (贪心算法-两个维度权衡题目)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: 168. Leetcode 134. 加
- 下一篇: 171. Leetcode 406. 根