24.二叉树和为某一值的路径

二叉树和为某一值的路径

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

题意中两个注意点：

1. 所有路径
2. 从树根结点到叶结点的路径，叶结点也就是终端结点。

DFS

所有路径结合二叉树我们想到了DFS，使用递归帮我们方便的解决问题。

我们按照前序遍历的方式，从二叉树的根结点开始

路径定为path，路径集合定为ans，目标值为target

• 在根结点不为NULL的情况下，开始递归
• 将根结点的值放入path
• 同时满足和为target、左结点为NULL、右结点为NULL三个条件则满足题意的路径，将path放入ans中
• 和小于等于target时且左结点不为NULL，递归左结点
• 和小于等于target时且右结点不为NULL，递归右结点
• 将根结点的值从path中移除（这一步非常重要，保证DFS过程中状态不影响其他状态。）
• 返回ans

/*
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> > ans;
    vector<int> path;
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if (root == NULL)
        {
            return ans;
        }
        path.push_back(root->val);
        if (root->left == NULL && root->right == NULL && expectNumber == root->val)
        {
            ans.push_back(path);
        }
        if (root->left != NULL && root->val <= expectNumber)
        {
            FindPath(root->left, expectNumber-root->val);
        }
        if (root->right != NULL && root->val <= expectNumber)
        {
            FindPath(root->right, expectNumber-root->val);
        }
        vector<int>::iterator end = path.end();
        path.erase(--end);
        return ans;
    }
};