剑指offer-二叉树中和为某一值的路径

描述

输入一颗二叉树的根节点和一个整数，按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

 

解决思路：

1. 注意这里的路径是根节点到叶子节点的遍历顺序，所以难度并不是很大（思路类似于dfs，单写起来比dfs更简单）
2. 其实遍历目标路径很简单，主要就路径记录的问题（dfs方便记录路径），我这里就用个成员变量记录了，有点小作弊的感觉。

class Solution {
private:
    vector<vector<int>> res;
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        // dfs吧 9         // 首先，怎么记录路径呢？11         // 用类成员变量？带检验路径需要逐层记录，所以需要另写递归函数
        if(root==nullptr){
            return res;
        }
        vector<int> cp;
        CalPath(root,expectNumber,cp);
        return res;    // 感觉有点像作弊
    }
    
    // 尽量用dfs，不然路径不好记录
    // 遍历到某节点时，如果总和大于aim，那么就返回空向量
    // 如果等于aim，就返回该向量
    // 如果小于aim，就继续遍历,并记录目前的路径
    void CalPath(TreeNode* root,int aim,vector<int> cp){
        cp.push_back(root->val);
        // 叶子节点作为递归出口
       if(root->left==nullptr && root->right==nullptr){ 
            if(root->val==aim){
                res.push_back(cp);
            }
            return;
        }
         // 非叶子节点
        if(root->val>=aim){   // 这里最好释放一下cp向量的栈空间
            return;
        }else{
            if(root->left!=nullptr) CalPath(root->left, aim-root->val, cp);
            if(root->right!=nullptr) CalPath(root->right, aim-root->val, cp);
        }
    }
};

 

如果是根节点到任意节点的路径，可采用如下代码（核心算法是正确的，但边界问题我没处理了）

 

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
private:
    vector<vector<int>> res;
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        // dfs吧17         // 这里根节点是固定，就不用考虑子树根节点了
        // 首先，怎么记录路径呢？
        // 遍历到下一层时，有的路径是正确，有些是不正确的
        // 用类成员变量？带检验路径需要逐层记录，所以需要另写递归函数
        vector<int> cp;
        CalPath(root,expectNumber,cp);
        return res;    // 感觉有点像作弊
    }
    
    // 尽量用dfs，不然路径不好记录
    // 遍历到某节点时，如果总和大于aim，那么就返回空向量
    // 如果等于aim，就返回该向量
    // 如果小于aim，就继续遍历,并记录目前的路径
    void CalPath(TreeNode* root,int aim,vector<int> cp){
        if(root==nullptr){
            return;
        }
        cp.push_back(root->val);
        if(root->val==aim){
            res.push_back(cp);
            return;
        }else if(root->val>aim){
            return;
        }else{
            CalPath(root->left, aim-root->val, cp);
            CalPath(root->right, aim-root->val, cp);
        }
    }
    
};