精品推荐:
《征服数据结构》专栏:50多种数据结构彻底征服。
《经典图论算法》专栏:50多种经典图论算法全部掌握。
今天在牛客网上看到一张图片,展示的是各互联网大厂的月薪分布,从分布结果来看,50K以上占比最高的是字节和拼多多。30~50k占比最高的是贝壳,其次是阿里,现在不是房价都降了吗,贝壳工资怎么还这么高?
图片来源:牛客网
--------------下面是今天的算法题--------------
来看下今天的算法题,这题是LeetCode的第113题:路径总和 II。
问题描述
来源:LeetCode第113题
难度:中等
给你二叉树的根节点 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
输出:[]
树中节点总数在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
问题分析
这题和昨天讲的路径总和类似,昨天讲的只要满足从根节点到叶子节点路径上的节点和等于targetSum即可。而这题是让找出所有从根节点到叶子节点路径和等于targetSum的路径。
我们从根节点开始,记录遍历路径上的所有节点,到叶子节点的时候如果路径上的节点和等于targetSum,就把这条路径记录下来。
注意路径的保存使用的是引用传递,最后还需要回溯,就是在递归往回走的时候把最后添加的给移除。
JAVA:
public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
List<List<Integer>> ans = new ArrayList<>();// 存放结果的集合
List<Integer> path = new ArrayList<>();// 存放从根节点到叶子节点的值
dfs(root, targetSum, path, ans);// dfs遍历
return ans;
}
public void dfs(TreeNode cur, int targetSum, List<Integer> path,
List<List<Integer>> ans) {
if (cur == null)// 节点为空,直接返回
return;
path.add(cur.val);// 把当前节点的值添加到路径path中
// 如果是叶子节点,当前路径上的和等于targetSum,就把当前路径添加进来
if (cur.left == null && cur.right == null && targetSum == cur.val) {
ans.add(new ArrayList<>(path));
} else {
// 沿着当前树的左右子节点往下遍历,这里要注意targetSum的值要减去当前节点的值
dfs(cur.left, targetSum - cur.val, path, ans);
dfs(cur.right, targetSum - cur.val, path, ans);
}
path.remove(path.size() - 1);// 回溯,把最后一个添加的节点给移除
}
C++:
public:
vector<vector<int>> pathSum(TreeNode *root, int targetSum) {
vector<vector<int>> ans;// 存放结果的集合
vector<int> path;// 存放从根节点到叶子节点的值
dfs(root, targetSum, path, ans);// dfs遍历
return ans;
}
void dfs(TreeNode *cur, int targetSum, vector<int> &path,
vector<vector<int>> &ans) {
if (!cur)// 节点为空,直接返回
return;
path.push_back(cur->val);// 把当前节点的值添加到路径path中
// 如果是叶子节点,当前路径上的和等于targetSum,就把当前路径添加进来
if (!cur->left && !cur->right && targetSum == cur->val) {
ans.push_back(path);
} else {
// 沿着当前树的左右子节点往下遍历,这里要注意targetSum的值要减去当前节点的值
dfs(cur->left, targetSum - cur->val, path, ans);
dfs(cur->right, targetSum - cur->val, path, ans);
}
path.pop_back();// 回溯,把最后一个添加的节点给移除
}
Python:
def pathSum(self, root: Optional[TreeNode], targetSum: int) -> List[List[int]]:
def dfs(cur, targetSum):
if not cur: # 节点为空,直接返回
return
path.append(cur.val) # 把当前节点的值添加到路径path中
# 如果是叶子节点,当前路径上的和等于targetSum,就把当前路径添加进来
if not cur.left and not cur.right and targetSum == cur.val:
ans.append(path[:])
else:
# 沿着当前树的左右子节点往下遍历,这里要注意targetSum的值要减去当前节点的值
dfs(cur.left, targetSum - cur.val)
dfs(cur.right, targetSum - cur.val)
path.pop() # 回溯,把最后一个添加的节点给移除
ans = [] # 存放结果的集合
path = [] # 存放从根节点到叶子节点的值
dfs(root, targetSum) # dfs遍历
return ans
笔者简介
博哥,真名:王一博,毕业十多年,《算法秘籍》作者,专注于数据结构和算法的讲解,在全球30多个算法网站中累计做题2000多道,在公众号中写算法题解800多题,对算法题有自己独特的解题思路和解题技巧,喜欢的可以给个关注,也可以下载我整理的1000多页的PDF算法文档。