二叉树的路径解析(任意两点/自顶向下)

写在前

本部分题目,讨论自顶向下的情况,就是从某一个节点(不一定是根节点),从上向下寻找路径,到某一个节点(不一定是叶节点)结束,而继续细分的话还可以分成一般路径与给定和的路径。
注意:这类题通常用深度优先搜索(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,因此一个点是不能构成路径的

巨人的肩膀

https://leetcode-cn.com/problems/smallest-string-starting-from-leaf/solution/yi-pian-wen-zhang-jie-jue-suo-you-er-cha-10sk/
b

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,056评论 5 474
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 84,842评论 2 378
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 148,938评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,296评论 1 272
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,292评论 5 363
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,413评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,824评论 3 393
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,493评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,686评论 1 295
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,502评论 2 318
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,553评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,281评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,820评论 3 305
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,873评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,109评论 1 258
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,699评论 2 348
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,257评论 2 341

推荐阅读更多精彩内容