写在前
本部分题目,讨论自顶向下的情况,就是从某一个节点(不一定是根节点),从上向下寻找路径,到某一个节点(不一定是叶节点)结束,而继续细分的话还可以分成一般路径与给定和的路径。
注意:这类题通常用深度优先搜索(DFS)和广度优先搜索(BFS)解决,BFS较DFS繁琐,这里为了简洁只展现DFS代码
1.二叉树的所有路径(257-易)
题目描述:给定一个二叉树,返回所有从根节点到叶子节点的路径。
示例:
输入: 1 / \ 2 3 \ 5 输出: ["1->2->5", "1->3"] 解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
思路:递归实现
- 如果root不为空,我们才进行拼接!
- 判断是否到达叶子节点。如果到达,加入集合返回结果;否则,拼接箭头,递归左右子树。
代码实现:
public List<String> binaryTreePaths(TreeNode root) { List<String> ans = new ArrayList<>(); if (root == null) { return ans; } dfs(root, "", ans); return ans; } private void dfs(TreeNode root, String path, List<String> ans) { if (root == null) { return; } StringBuilder sb = new StringBuilder(path); sb.append(String.valueOf(root.val)); if (root.left == null && root.right == null) { ans.add(sb.toString()); } else { sb.append("->"); dfs(root.left, sb.toString(), ans); dfs(root.right, sb.toString(), ans); } }
2.求和路径(面试题04.12)
题目描述:给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树,以及目标和 sum = 22, 5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1 返回:3 解释:和为 22 的路径有:[5,4,11,2], [5,8,4,5], [4,11,7]
思路:递归实现
- 本题关键是路径的起止点不确定,所以我们要在主函数递归左右子节点(累加结果)
- 我们维护一个变量记录结果,即当root.val == sum时我们获得一个结果,累积左右子节点。
代码实现:
public int pathSum(TreeNode root, int sum) { if (root == null) { return 0; } return dfs(root, sum) + pathSum(root.left, sum) + pathSum(root.right, sum); } private int dfs(TreeNode root, int sum) { if (root == null) { return 0; } int ans = sum == root.val ? 1 : 0; sum -= root.val; return dfs(root.left, sum) + dfs(root.right, sum) + ans; }
3.路径总和(112-易)
题目描述:给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
示例:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22 输出:true
思路:递归实现
- 终止条件,如果当前节点为空,返回false
- 到达叶子节点,且sum = root.val
- 递归的左右子节点(注意更新sum值,即减去root.val)
代码实现:
public boolean hasPathSum(TreeNode root, int sum) { if (root == null) { return false; } sum -= root.val; if (root.left == null && root.right == null && sum == 0) { return true; } return hasPathSum(root.left, sum) || hasPathSum(root.right, sum); }
4.路径总和II(113-中)
题目描述:本题在上题基础上(从根节点到叶子节点的路径和等于给定值),要求找出这些路径!
示例:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22 输出:[[5,4,11,2],[5,8,4,5]]
思路:递归实现,注意本题与257不同的是需要回溯。
- 我们先加入路径,不满足需要从队列中删除,所以需要实现一个双端队列。
- 注意找到一个结果也需要继续进行回溯。
代码实现:
List<List<Integer>> ans = new LinkedList<>(); Deque<Integer> path = new LinkedList<>(); public List<List<Integer>> pathSum(TreeNode root, int targetSum) { if (root == null) { return ans; } dfs(root, targetSum); return ans; } public void dfs(TreeNode root, int targetSum) { if (root == null) { return; } path.offerLast(root.val); targetSum -= root.val; if (root.left == null && root.right == null && targetSum == 0) { ans.add(new LinkedList<>(path)); } dfs(root.left, targetSum); dfs(root.right, targetSum); path.pollLast(); }
拓展(T437):这里路径不限制从根节点到叶子节点,但是方向一定是从上到下的。
- 注意:区分开始节点(主递归函数)和结束节点(判断结束标志)!
代码实现:
public int pathSum(TreeNode root, int sum) { if (root == null) { return 0; } return pathSumStartWithRoot(root, sum) + pathSum(root.left, sum) + pathSum(root.right, sum); } private int pathSumStartWithRoot(TreeNode root, int sum) { if (root == null) { return 0; } int ans = 0; sum -= root.val; if (sum == 0) { ans++; } ans += pathSumStartWithRoot(root.left, sum) + pathSumStartWithRoot(root.right, sum); return ans; }
5.从叶子节点开始的最小字符串(988-中)
题目描述:给定一颗根结点为 root 的二叉树,树中的每一个结点都有一个从 0 到 25 的值,分别代表字母 'a' 到 'z':值 0 代表 'a',值 1 代表 'b',依此类推。
找出按字典序最小的字符串,该字符串从这棵树的一个叶结点开始,到根结点结束。
(小贴士:字符串中任何较短的前缀在字典序上都是较小的:例如,在字典序上 "ab" 比 "aba" 要小。叶结点是指没有子结点的结点。)
示例:
输入:[0,1,2,3,4,3,4] 输出:"dba"
思路:递归实现,需要进行回溯,一些必要的函数:
- 直接使用StringBuilder的reverse()方法进行反转,反转两次是为了继续进行拼接。
- 使用compareTo()比较两个字符串的字典序(本质是比较ascii值)
- 使用deleteCharAt(待删除的索引)进行回溯
注意:我们是更新获取最小的字典序,所以我们初始值应该大于'z'的ascii值
代码实现:
String ans = "~"; public String smallestFromLeaf(TreeNode root) { if (root == null) { return ans; } dfs(root, new StringBuilder()); return ans; } private void dfs(TreeNode root, StringBuilder sb) { if (root == null) { return; } sb.append((char)(root.val + 'a')); if (root.left == null && root.right == null) { sb.reverse(); String tmp = sb.toString(); sb.reverse(); if (tmp.compareTo(ans) < 0) { ans = tmp; } } dfs(root.left, sb); dfs(root.right, sb); sb.deleteCharAt(sb.length() - 1); }
6.根节点到叶子节点数字之和(129-中)
题目描述:给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。
每条从根节点到叶节点的路径都代表一个数字:
例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。计算从根节点到叶节点生成的 所有数字之和 。
示例:
输入:root = [1,2,3] 输出:25 解释: 从根到叶子节点路径 1->2 代表数字 12 从根到叶子节点路径 1->3 代表数字 13 因此,数字总和 = 12 + 13 = 25
思路:递归实现,递归所有的路径(根节点到叶子节点),进行累加。
- 定义一个变量记录自上向下的路径和。
- 当到达叶子节点,找到一条路径和。
- 递归的寻找左右子节点。
代码实现:
public int sumNumbers(TreeNode root) { return dfs(root, 0); } private int dfs(TreeNode root, int pathSum) { if (root == null) { return 0; } pathSum = pathSum * 10 + root.val; if (root.left == null && root.right == null) { return pathSum; } return dfs(root.left, pathSum) + dfs(root.right, pathSum); }
注意点
- 如果是找路径和等于给定target的路径的,那么可以不用新增一个临时变量cursum来判断当前路径和, 只需要用给定和target减去节点值,最终结束条件判断target==0即可。
- 是否要回溯:二叉树的问题大部分是不需要回溯的,原因如下:二叉树的递归部分:dfs(root->left),dfs(root->right)已经把可能的路径穷尽了,因此到任意叶节点的路径只可能有一条,绝对不可能出现另外的路径也到这个满足条件的叶节点的;但是对路径有限制,则需要使用回溯!如T4和T5;
二维数组(例如迷宫问题)的DFS,for循环向四个方向查找每次只能朝向一个方向,并没有穷尽路径, 因此某一个满足条件的点可能是有多条路径到该点的;
并且visited数组标记已经走过的路径是会受到另外路径是否访问的影响,这时候必须回溯 - 找到路径后是否要return:取决于题目是否要求找到叶节点满足条件的路径,如果必须到叶节点,那么就要return(也可以不return); 但如果是到任意节点都可以,那么必不能return,因为这条路径下面还可能有更深的路径满足条件,还要在此基础上继续递归
- 是否要双重递归(即调用根节点的dfs函数后,继续调用根左右节点的pathsum函数):看题目要不要求从根节点开始的,还是从任意节点开始
本部分题目,非自顶向下:就是从任意节点到任意节点的路径,不需要自顶向下
7.二叉树最大路径和(124-难)
题目描述:路径 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和 是路径中各节点值的总和。给你一个二叉树的根节点 root ,返回其 最大路径和 。
示例:
输入:root = [-10,9,20,null,null,15,7] 输出:42 解释:最优路径是 15 -> 20 -> 7 ,路径和为 15 + 20 + 7 = 42
思路:递归实现,递归函数的关键写好递归处理的逻辑,怎么处理当前子树,需要返回什么,设置递归出口。关键是正确定义递归函数。
- 递归函数:计算当前节点为父亲节点提供的贡献(即当前节点到根节点的最大路径)。
- 递归的计算左右子节点的最大贡献值,逻辑:贡献值大于0,才会选取对应的节点最大贡献值
- 更新最大路径和(当前节点值和左右子节点最大贡献值)
特别注意:dfs函数中定义的是一个节点能贡献的最大值(计算该节点到根节点的最大路径),而我们统计结果时更新的二叉树中任意两点的最大路径(可能是多个节点到这个根节点的组合)!
代码实现:
private int ans = 0; public int longestUnivaluePath(TreeNode root) { dfs(root); return ans; } private int dfs(TreeNode root) { if (root == null) { return 0; } int l = dfs(root.left); int r = dfs(root.right); int left = root.left != null && root.val == root.left.val ? l : 0; int right = root.right != null && root.val == root.right.val ? r : 0; ans = Math.max(ans, left + right); return Math.max(left, right) + 1; }
8.最长同值路径(687-中)
题目描述:给定一个二叉树,找到最长的路径,这个路径中的每个节点具有相同值。 这条路径可以经过也可以不经过根节点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例:
5 / \ 4 5 / \ \ 1 1 5 返回 2
思路:递归实现,思想与上题基本相似。
- 递归函数:当前节点到根节点的最长同值路径
- 递归左右子树的最长同值路径;逻辑:出现同值,更新路径左右路径,否则为0
- 更新最长同值路径
代码实现:
public int pathSum(TreeNode root, int sum) { if (root == null) { return 0; } return dfs(root, sum) + pathSum(root.left, sum) + pathSum(root.right, sum); } private int dfs(TreeNode root, int sum) { if (root == null) { return 0; } int ans = sum == root.val ? 1 : 0; sum -= root.val; return dfs(root.left, sum) + dfs(root.right, sum) + ans; }
9.二叉树的直径(543-易)
题目描述:给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
注意:两个节点之间的路径长度由它们之间的边数表示。
示例:
给定二叉树 1 / \ 2 3 / \ 4 5 返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
思路:递归实现,递归实现,思想与上题基本相似,还是区分递归函数和最终结果的区别。
代码实现:
private int max = 0; public int diameterOfBinaryTree(TreeNode root) { dfs(root); return max; } private int dfs(TreeNode root) { if (root == null) { return 0; } int left = dfs(root.left); int right = dfs(root.right); max = Math.max(max, left + right); return Math.max(left, right) + 1; }
附(特殊):所有距离为k的节点(863-中)
题目描述:给定一个二叉树(具有根结点 root), 一个目标结点 target ,和一个整数值 K 。
返回到目标结点 target 距离为 K 的所有结点的值的列表。 答案可以以任何顺序返回。
示例:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2 输出:[7,4,1] 解释: 所求结点为与目标结点(值为 5)距离为 2 的结点, 值分别为 7,4,以及 1
思路:本题搜索的大体思路可以按照图的bfs,我们需要给所有节点添加一个指向父节点的引用(这样就知道了距离该节点1距离的所有节点),进而找到距离target节点为k的所有节点。
- 用map集合构建每个节点的父节点,即建立当前节点到父节点的引用(构建图)
- 用hashset记录节点是否被访问过
- 我们可以看成以当前节点为中心的辐射(是一种广度搜索的思想),完成一层之后k--。
代码实现:
private Map<TreeNode, TreeNode> parents; private Set<TreeNode> isVisited; private List<Integer> ans; public List<Integer> distanceK(TreeNode root, TreeNode target, int k) { parents = new HashMap<>(); getParents(root, null); isVisited = new HashSet<>(); ans = new ArrayList<>(); Deque<TreeNode> queue = new LinkedList<>(); queue.add(target); while (!queue.isEmpty()) { if (k-- == 0) { while (!queue.isEmpty()) { ans.add(queue.poll().val); } break; } int size = queue.size(); for (int i = 0; i < size; ++i) { TreeNode node = queue.poll(); isVisited.add(node); if (node.left != null && !isVisited.contains(node.left)) { queue.add(node.left); } if (node.right != null && !isVisited.contains(node.right)) { queue.add(node.right); } TreeNode p = parents.get(node); if (p != null && !isVisited.contains(p)) { queue.add(p); } } } return ans; } private void getParents(TreeNode root, TreeNode parent) { if (root == null) { return; } parents.put(root, parent); getParents(root.left, root); getParents(root.right, root); }
注意点
- left,right代表的含义要根据题目所求设置,比如最长路径、最大路径和等等
- 全局变量res的初值设置是0还是INT_MIN要看题目节点是否存在负值,如果存在就用INT_MIN,否则就是0
- 注意两点之间路径为1,因此一个点是不能构成路径的