[剑指Offer]二叉树中结点值的和为输入整数的所有路径

二叉树指定值

Posted by JinFei on January 18, 2020

题目描述

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

解题思路

  • 用一个vector代表当前路径
  • 检查当前节点是否是叶子节点
  • 如果是叶子节点,判断是否等于给定的值
  • 如果不是,递归的查找
  • 返回时,记得把当前vector的元素给弹出
/*
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) {
        vector<vector<int> > res;
        vector<int> path;
        if(root == NULL){
            return res;
        }
        
        FindPath(root, expectNumber, 0, res, path);
        return res;
    }
    void FindPath(TreeNode* root, int expectNumber, int currentNumber, vector<vector<int> >& res, vector<int>& path){
        currentNumber += root -> val;
        path.push_back(root -> val);
        bool isleaf = root -> left == NULL && root -> right == NULL;
        if(isleaf && currentNumber == expectNumber){
            res.push_back(path);
        }
        if(root -> left){
            FindPath(root -> left, expectNumber, currentNumber, res, path);
        }
        if(root -> right){
            FindPath(root -> right, expectNumber, currentNumber, res, path);
        }
        path.pop_back();
    }
};

0118解题思路

  • 路径pop的时候 有点问题
  • 遍历完左右子树之后,才需要pop
  • 而不是不相等了 就要pop一下
  • 需要走到叶子节点后,不相等(想一想最后执行pop_back,一定是走到叶子节点后,发现不等了,才会pop_back操作)