剑指offer系列24之二叉树中和为某一值的路径

Cheng Qian

于 2019-07-03 19:02:01 发布

阅读量101

点赞数

分类专栏：剑指offer python coding

本文链接：https://blog.csdn.net/xiayangmian4130/article/details/88088809

版权

coding 同时被 3 个专栏收录

35 篇文章 0 订阅

订阅专栏

剑指offer

34 篇文章 0 订阅

订阅专栏

python

15 篇文章 0 订阅

订阅专栏

要求

时间限制：1秒
空间限制：32768K
热度指数：288594

题目描述

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

思路

这类问题可以用带记忆的DFS来解决。

python实现

# -*- coding:utf-8 -*-
# -*- coding:utf-8 -*-
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        # write code here
        self.res = []
        self.path = []
        self.findPathCore(root, expectNumber)
        return self.res

    def findPathCore(self, root, expectNumber):
        if not root:
            return
        self.path.append(root.val)
        if root.val == expectNumber and not root.left and not root.right:
            self.res.append([e for e in self.path])
        else:
            if root.left:
                self.findPathCore(root.left, expectNumber - root.val)
            if root.right:
                self.findPathCore(root.right, expectNumber - root.val)
        self.path.pop(-1)

C++实现

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<vector<int> > res;
    vector<int> path;
    void dfs(TreeNode* root, int number){
        if (!root) return;
        path.push_back(root->val);
        if (!root->left && !root->right && root->val==number)
            res.push_back(path);
        else{
            if (root->left)
                dfs(root->left, number-root->val);
            if (root->right)
                dfs(root->right, number-root->val);
        }
        // 遇到满足条件的叶节点下一步就要弹出这个叶节点回到其父节点
        path.pop_back();
    }
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        dfs(root, expectNumber);
        return res;
    }
};