文章目录
一,路径总和 II
1,程序简介
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
- 输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
- 输出:[[5,4,11,2],[5,8,4,5]]
示例 2:
- 输入:root = [1,2,3], targetSum = 5
- 输出:[]
示例 3:
- 输入:root = [1,2], targetSum = 0
- 输出:[]
提示:
- 树中节点总数在范围 [0, 5000] 内
- -1000 <= Node.val <= 1000
- -1000 <= targetSum <= 1000
2,程序代码
# -*- coding: utf-8 -*-
"""
Created on Tue Feb 22 09:28:22 2022
Function: 路径总和 II
@author: 小梁aixj
"""
class Solution:
def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
pathvalue = 0
path = []
result = []
def preorder(node, pathvalue, sum, path, result):
if node == None:
return
pathvalue += node.val
path.append(node.val)
if pathvalue == sum and node.left == None and node.right == None:
result.append(list(path)) # 注意加list
preorder(node.left, pathvalue, sum, path, result)
preorder(node.right, pathvalue, sum, path, result)
pathvalue -= node.val
path.pop()
preorder(root, pathvalue, sum, path, result)
return result
二,分发糖果
1,程序简介
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
- 每个孩子至少分配到 1 个糖果。
- 评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
- 输入:[1,0,2]
- 输出:5
- 解释:你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
- 输入:[1,2,2]
- 输出:4
- 解释:你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
2,程序代码
# -*- coding: utf-8 -*-
"""
Created on Tue Feb 22 09:29:46 2022
Function: 分发糖果
@author: 小梁aixj
"""
class Solution(object):
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: int
"""
if ratings == None:
return 0
lenth = len(ratings)
dp = [1 for i in range(lenth)]
sum = 0
for i in range(1, len(ratings)):
if ratings[i] > ratings[i - 1]:
dp[i] = dp[i - 1] + 1
for i in range(len(ratings) - 2, -1, -1):
if ratings[i] > ratings[i + 1] and dp[i] <= dp[i + 1]:
dp[i] = dp[i + 1] + 1
for i in range(lenth):
sum += dp[i]
return sum
三,存在重复元素
1,程序简介
给定一个整数数组,判断是否存在重复元素。
如果存在一值在数组中出现至少两次,函数返回 true 。如果数组中每个元素都不相同,则返回 false 。
示例 1:
- 输入: [1,2,3,1]
- 输出: true
示例 2:
- 输入: [1,2,3,4]
- 输出: false
示例 3:
- 输入: [1,1,1,3,3,4,3,2,4,2]
- 输出: true
2,程序代码
# -*- coding: utf-8 -*-
"""
Created on Tue Feb 22 09:30:11 2022
Function: 存在重复元素
@author: 小梁aixj
"""
class Solution:
def containsDuplicate(self, nums: List[int]) -> bool:
nums.sort()
count = 0
while count < len(nums) - 1:
if nums[count] == nums[count + 1]:
return True
count += 1
return False