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

时间限制:1秒空间限制:32768K热度指数:8801

**算法知识视频讲解

题目描述

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

题解:利用dfs

代码如下:

/*
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<int>v;
        int sum=0;
        vector<vector<int> >result;
        dfs(root,result,v,sum,expectNumber,0);
        return result;
    }
    
    void dfs(TreeNode* root,vector<vector<int> > &result,vector<int>&v,int &k,int expectNumber,int flag){
        if(root==NULL){
            return;
        }
        k+=root->val;
        v.push_back(root->val);
        if(root->left==NULL&&root->right==NULL&&k==expectNumber){
            result.push_back(v);
            k-=root->val;
            v.pop_back();
            if(flag==1){
                k-=v[v.size()-1];
                v.pop_back();
            }
            return;
        }else if(root->left==NULL&&root->right==NULL&&k!=expectNumber){
            k-=root->val;
            v.pop_back();
            return;
        }
        dfs(root->left,result,v,k,expectNumber,2);
        dfs(root->right,result,v,k,expectNumber,1);
    }
};

打赏一个呗

取消

感谢您的支持,我会继续努力的!

扫码支持
扫码支持
扫码打赏,你说多少就多少

打开支付宝扫一扫,即可进行扫码打赏哦