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

  1. 题目:输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
  2. 思路:
    1. DFS遍历,到叶子结点之后检查路径和是否为sum,如果是sum加入结果集
  3. 启发或者坑:
    1. 递归变量可以使用引用,极大节省空间
  4. 代码
    /*
    struct TreeNode {
        int val;
        struct TreeNode *left;
        struct TreeNode *right;
        TreeNode(int x) :
                val(x), left(NULL), right(NULL) {
        }
    };*/
     
     
    class Solution {
    public:
        int eN;
        vector<vector<int>> res;
        vector<vector<int>> FindPath(TreeNode* root, int expectNumber) {
            if (!root)
                return res;
            eN = expectNumber;
            vector<int> temp;
            int target = 0;
            FindAllPath(root, temp, target);
            return res;
        }
     
        void FindAllPath(TreeNode* root, vector<int>& path, int& sum) {
            //这个路径指的是从根节点到叶子几点的路径,遍历所有叶子节点的路径,如果等于eN,输出其路径
             
            path.push_back(root->val);
            sum += root->val;
            cout <<sum << endl;
            if (!root->left && !root->right) {
                if (sum == eN)
                    res.push_back(path);     
                return;
            }
            if (root->left) {
                FindAllPath(root->left, path, sum);
                //cout << "start new leaf " <<endl;
                //cout << "before:" << sum << endl;
                sum -= path[path.size()-1];
                path.pop_back();
                //cout << "AFTER:" << sum << endl;
            }
            if (root->right) {
                FindAllPath(root->right, path, sum);
                sum -= path[path.size()-1];
                path.pop_back();
            }   
        }
    };
发布了131 篇原创文章 · 获赞 5 · 访问量 7402

猜你喜欢

转载自blog.csdn.net/Alexia23/article/details/103843129