当前位置:  首页>> 技术小册>> 数据结构与算法(下)

题目描述

输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

解法

  1. import java.util.ArrayList;
  2. /**
  3. * @author bingo
  4. * @since 2018/11/23
  5. */
  6. /**
  7. public class TreeNode {
  8. int val = 0;
  9. TreeNode left = null;
  10. TreeNode right = null;
  11. public TreeNode(int val) {
  12. this.val = val;
  13. }
  14. }
  15. */
  16. public class Solution {
  17. private ArrayList<ArrayList<Integer>> res = new ArrayList<>();
  18. /**
  19. * 找出二叉树中和为某一值的路径(必须从根节点到叶节点)
  20. *
  21. * @param root 二叉树的根结点
  22. * @param target 目标值
  23. * @return 结果list
  24. */
  25. public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
  26. findPath(root, target, new ArrayList<>());
  27. return res;
  28. }
  29. private void findPath(TreeNode root, int target, ArrayList<Integer> list) {
  30. if (root == null) {
  31. return;
  32. }
  33. list.add(root.val);
  34. target -= root.val;
  35. if (target == 0 && root.left == null && root.right == null) {
  36. res.add(new ArrayList<>(list));
  37. } else {
  38. findPath(root.left, target, list);
  39. findPath(root.right, target, list);
  40. }
  41. list.remove(list.size() - 1);
  42. }
  43. }

测试用例

  1. 功能测试(二叉树中有一条、多条符合要求的路径;二叉树中没有符合要求的路径);
  2. 特殊输入测试(指向二叉树根节点的指针为空指针)。

该分类下的相关小册推荐: