480. 二叉树的所有路径

从物理学到计算机,再到硬件,再到人工智能!
蓝桥杯备赛 (LintCode上刷的第16题)

问题描述

给一棵二叉树,找出从根节点到叶子节点的所有路径。

样例输出

在这里插入图片描述

JAVA代码实现

一共有两种方法,一种是分治法,另一种是遍历的方法。

package tree;

import java.util.ArrayList;
import java.util.List;

public class BinaryTreePaths1230 {

	/**
	 * Divide Conquer : 分治法
	 * @param root the root of the binary tree
	 * @return all root-to-leaf
	 */
	public List<String> binaryTreePaths1(TreeNode1230 root) {
		//创建一个数列记录路径path
		List<String> paths = new ArrayList<>();
		//判断根为空的情况
		if (root == null) {
			return paths;
		}
		//左子节点
		List<String> leftPaths = binaryTreePaths1(root.left);
		//右子节点
		List<String> rightPaths = binaryTreePaths1(root.right);
		//遍历左子树
		for (String path : leftPaths) {
			paths.add(root.val + "-> " + path);
		}
		//遍历右子树
		for (String path : rightPaths) {
			paths.add(root.val + "-> " + path);
		}
		//paths长度为0,则只有根可以添加进路径数列
		if(paths.size() == 0) {
			paths.add("" + root.val);
		}
		return paths;
	}
	
	/**
	 * Traverse : 遍历算法
	 * @param root the root of the binary tree
	 * @return all root-to-leaf
	 */
	public List<String> binaryTreePaths2(TreeNode1230 root) {
		//创建一个数列记录结果
		List<String> result = new ArrayList<>();
		//判断根是否为空
		if (root == null) {
			return result;
		}
		helper(root, String.valueOf(root.val), result);
		return result;
	}
	
	/**
	 * 
	 * @param root  the root of the binary tree 
	 * @param path  需要加入路径数列的某个节点的值
	 * @param result 记录路径结果的数列
	 */
	public void helper(TreeNode1230 root, String path, List<String> result) {
		if (root == null) {
			return;
		}
		//只有根一个节点,将根加入路径
		if (root.left == null && root.right == null) {
			result.add(path);
			return;
		}
		//查找左子树
		if (root.left != null) {
			helper(root.left, path + "->" + String.valueOf(root.left), result);
		}
		//查找右子树
		if (root.right != null) {
			helper(root.right, path + "->" + String.valueOf(root.right), result);
		}
	}
}

参考九章算法

猜你喜欢

转载自blog.csdn.net/qq_43269495/article/details/85383521