题目:
目录
解法一:深度优先遍历
采用深度优先遍历很简单,只需要在深度优先遍历的基础上加以判断,若当前节点是叶子节点就将其路径添加到list集合中,需要传入String类型的参数用来更新记录路径。
注意:在递归函数中,我们传入例如引用类型的list和基本类型str作为参数,需注意两者的区别
如果将Java引用类型变量作为参数传递给方法,是将引用变量的值传递给形参,而引用变量的值实际上就是引用对象在堆内存中的地址。也就是说,这个时候实参和形参指向了同一个对象,如果利用形参进行操作,操作的就是实参指向的对象,最后通过实参的那个引用访问,自然是被形参操作过的结果。
如下:
public class test01{
public static void main(String[] args) {
List<String> list = new ArrayList<>();
String str = "a";
test(list,str);
}
//变量参数和引用参数递归测试
public static void test(List<String> list,String n){
if (list.size() > 5){
return;
}
list.add(n);
test(list,n + "a");
System.out.print(list);
System.out.println("----" + n);
}
}
运行结果为:
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----aaaaaa
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----aaaaa
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----aaaa
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----aaa
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----aa
[a, aa, aaa, aaaa, aaaaa, aaaaaa]----a
代码实现:
public List<String> binaryTreePaths(TreeNode root) {
List<String> list = new ArrayList<String>();
dfs(root,list,"");
return list;
}
public void dfs(TreeNode root,List<String> list,String str){
//递归终止条件
if(root == null){
return;
}
//遍历到叶子节点加入list中
if(root.left == null && root.right == null){
list.add(str + root.val);
}
//递归左右子树,在递归子树之前将上一节点的路径作为参数传入
dfs(root.left,list,str + root.val + "->");
dfs(root.right,list,str + root.val + "->");
}
解法二:广度优先遍历
使用广度优先遍历比较麻烦的是,他的遍历顺序是层序遍历,我们不能像深度优先遍历那样通过递归遍历直接得到路径,这时我们可以维护两个队列(nodeQueue,pathQueue)来实现,通过pathQueue来对应nodeQueue,将nodeQueue中每个节点的路径都对应存放在pathQueue中,来实现每个节点路径的独立性。
代码实现:
public List<String> binaryTreePaths(TreeNode root){
//创建list来存放叶子节点路径
List<String> list= new ArrayList<String>();
if(root == null){
return list;
}
//定义nodeQueue和pathQueue
Queue<TreeNode> nodeQueue = new LinkedList<TreeNode>();
Queue<String> pathQueue = new LinkedList<String>();
//同时对nodeQueue和pathQueue进行添加
nodeQueue.offer(root);
pathQueue.offer(Integer.toString(root.val));
while(!nodeQueue.isEmpty()){
//同时poll
TreeNode node = nodeQueue.poll();
String path = pathQueue.poll();
//判断node是不是叶子节点
if(node.left == null && node.right == null){
list.add(path);
}
if(node.left != null){
nodeQueue.offer(node.left);
pathQueue.offer(new StringBuilder(path).append("->" + node.left.val).toString());
}
if(node.right != null){
nodeQueue.offer(node.right);
pathQueue.offer(new StringBuilder(path).append("->" + node.right.val).toString());
}
}
return list;
}