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

剑指offer题型分类及各题的代码及解题思路

1、题目描述

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

2、初始结构定义如下：

/*
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> > FindPath(TreeNode* root,int expectNumber) 
    {}
};

3、思路

非递归法：后序遍历

① 进栈时候，把值同时压入路径的向量数组，修正路径的和

② 出栈时候，先判断和是否相等，且该节点是否是叶节点，

判断完成后保持和栈一致，抛出路径，修改路径的和

③ 向量数组和栈的操作要保持一致

4、C++实现代码如下

class Solution 
{
public:
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) 
    {
        vector<vector<int>> res;
        stack<TreeNode*> s;
        vector<int> v;
        
        //由根节点遍历至叶结点，将结点指针入栈，结点值放入vector，之后判断expectNumber是否为0，
        //若为0，即为所求，然后回退到目前结点的父节点，若右子树不为空，则遍历右子树，
        //则说明根节点的左子树已经遍历完了，将root置空，结束内部循环
        while(root || !s.empty())
        {
            while(root)
            {
                s.push(root);
                v.push_back(root->val);
                expectNumber -= root->val;
                root = root->left ? root->left : root->right;
            }
            //保存栈顶指针
            root = s.top();
            //此时路径值之和等于输入整数且遍历到叶结点了，其即为所求
            if(expectNumber == 0 && root->left == NULL && root->right == NULL)
            {
                res.push_back(v);
            }
            //栈顶指针出栈，vector尾删一个元素，exceptNumber加上删的元素值
            s.pop();
            v.pop_back();
            expectNumber += root->val;
            //栈不为空且当前栈顶结点指针右结点不为空，遍历右子树，否则root置空，强制出栈
            if(!s.empty() && s.top()->left == root)
            {
                root = s.top()->right;
            }
            else
                root = NULL;
        }
        return res;
    }
};