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

题目描述

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

 

解：将二叉树看成图，使用DFS进行遍历，遍历时进行数据保存和回溯。

 

struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};

class Solution {
public:
    vector<int> temp;
    vector<vector<int> > res;
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if(root)
            CheckPath(root, expectNumber);
        return res;
    }

    void CheckPath(TreeNode *root, int key)
    {
        temp.push_back(root ->val);
        if(key - root ->val == 0 && root -> left == NULL && root -> right == NULL)
            res.push_back(temp);
        else
        {
            if(root -> left != NULL)
                CheckPath(root -> left, key - root -> val);
            if(root -> right != NULL)
                CheckPath(root -> right, key - root -> val);
        }
        temp.pop_back();
    }
};