二叉树递归java

时间:2018-01-21 03:34:09

标签: java string recursion binary-tree

我想编写一个函数,它返回一个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处停止。我该怎么办?

1 个答案:

答案 0 :(得分:1)

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

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

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

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";
}