使用递归和回溯法找二叉树的所有路径

使用递归和回溯法找二叉树的所有路径

题目是 257.二叉树的所有路径

这道题需要我们找到所有从根节点到叶子结点的路径,起初我想的是使用深度优先遍历,但是我在遍历的时候没有找到方法来保存之前已经遍历过的节点。
按照代码随想录的解法,应该用一个List path保存之前已经遍历过的节点,当遍历到一个叶子节点就停止遍历,然后本轮递归的上一层递归把path的最后一个元素删掉。
如果当前节点不是叶子节点,就在递归当前节点的子节点后使用回溯,回溯的操作是删除当前节点的直接子节点,也就是path的最后一个元素,为什么只需要删除最后一个元素,因为在递归过程中已经删除了当前节点的其他子节点。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private void myPath(List<String> list, List<Integer> path, TreeNode root){
        path.add(root.val);
        if(root.left==null && root.right==null){
            StringBuilder sb=new StringBuilder();
            for(int i=0; i<path.size()-1; i++){
                sb.append(path.get(i));
                sb.append("->");
            }
            sb.append(path.get(path.size()-1));
            list.add(sb.toString());
            return;
        }
        if(root.left!=null){
            myPath(list, path, root.left);
            //回溯到当前的节点
            path.remove(path.size()-1);
        }
        if(root.right!=null){
            myPath(list, path, root.right);
            //回溯到当前的节点
            path.remove(path.size()-1);
        }
    }
    public List<String> binaryTreePaths(TreeNode root) {
        StringBuilder sb;
        List<String> list=new ArrayList<>();
        List<Integer>path=new ArrayList<>();
        myPath(list,path,root);
        return list;
    }
}

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注