二叉树递归 java
Binary tree recursion java
我想编写一个函数,其中 return 是一个由 0 和 1 组成的字符串,表示从根节点到特定节点 p 的路径,其中 0 表示向左走,1 表示向右走。
__3__
/ \
4 5
/ \ /
9 2 1
2 应该 return "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";
}
这是我对递归方式的想法,但是当它到达叶子时它只是 returns -1,而我希望它继续递归。比如,要找到 2,它会尝试 3 -> 4-> 9,但不会继续 4 的递归,而是在 9 处停止。我该怎么办?
您将返回一个特殊的 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";
}
我想编写一个函数,其中 return 是一个由 0 和 1 组成的字符串,表示从根节点到特定节点 p 的路径,其中 0 表示向左走,1 表示向右走。
__3__
/ \
4 5
/ \ /
9 2 1
2 应该 return "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";
}
这是我对递归方式的想法,但是当它到达叶子时它只是 returns -1,而我希望它继续递归。比如,要找到 2,它会尝试 3 -> 4-> 9,但不会继续 4 的递归,而是在 9 处停止。我该怎么办?
您将返回一个特殊的 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";
}