472 · 二叉树的路径和 III

最新推荐文章于 2024-08-30 16:28:36 发布

阿飞算法

最新推荐文章于 2024-08-30 16:28:36 发布

阅读量107

点赞数

分类专栏： Lintcode

本文链接：https://blog.csdn.net/wat1r/article/details/118692039

版权

Lintcode 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

472 · 二叉树的路径和 III

先中序遍历获取每一个节点
做dfs，以每个节点做为起点开始做回溯

方法1：DFS

List<List<Integer>> res = new ArrayList<>();

        public List<List<Integer>> binaryTreePathSum3(ParentTreeNode root, int target) {
            if (root == null) return res;
            List<ParentTreeNode> nodeList = new ArrayList<>();
            inOrder(root, nodeList);
            for (ParentTreeNode node : nodeList) {
                Set<ParentTreeNode> set = new HashSet<>();
                dfs(node,set,new ArrayList<>(),target);
            }
            return res;
        }


        private void inOrder(ParentTreeNode root, List<ParentTreeNode> nodeList) {
            if (root == null) return;
            inOrder(root.left, nodeList);
            nodeList.add(root);
            inOrder(root.right, nodeList);
        }

        private void dfs(ParentTreeNode root, Set<ParentTreeNode> set, List<Integer> path, int target) {
            if (root == null || set.contains(root)) return;
            set.add(root);
            path.add(root.val);
            target -= root.val;
            if (target == 0) {
                res.add(new ArrayList<>(path));
            }
            dfs(root.left, set, path, target);
            dfs(root.right, set, path, target);
            dfs(root.parent, set, path, target);
            set.remove(root);
            path.remove(path.size() - 1);
            target += root.val;
        }