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