二叉树递归java

Question

我想编写一个函数，它返回一串 0 和 1 来表示从根节点到特定节点 p 的路径，其中 0 表示向左，1 表示向右。

    __3__
   /     \
  4       5 
 / \     / 
9   2    1

2 应该返回“01”

public String traverse(TreeNode root, TreeNode p,String s) {
    if (root == p) return s;
    if (root.left != null) return traverse(root.left,p,s + "0");

    if (root.right != null) return traverse(root.right,p,s+ "1"); 

    return "-1";
  }

这是我对递归方式的想法，但是当它到达叶子时它只返回-1，而我希望它继续递归。如，要找到 2，它会尝试 3 -> 4-> 9，但不会继续 4 的递归，而是在 9 处停止。我该怎么办？

Answer 1

当您在树的特定分支上找不到您要查找的值时，您将返回一个特殊的 sentinel 值“-1”。

但是您没有在代码中的任何地方检查此值 - 当您要查找的值不在树的左分支上时，您需要在右分支中搜索。

因此，您需要在搜索树的左分支后添加一个检查，即您没有获得哨兵值作为结果。如果你这样做了，你需要继续搜索。

public String traverse(TreeNode root, TreeNode p, String s) {
    if (root == p) {
        return s;
    }
    if (root.left != null) {
        String result = traverse(root.left, p, s + "0");
        if (!"-1".equals(result)) {
            return result;
        }
    }
    if (root.right != null) {
        return traverse(root.right, p, s + "1");
    }
    return "-1";
}