Leetcode刷题笔记——剑指 Offer 34. 二叉树中和为某一值的路径【DFS】

题目

Leetcode刷题笔记——剑指 Offer 34. 二叉树中和为某一值的路径【DFS】_第1张图片

思路

显然这是一道DFS+回溯题,在DFS中加回溯需要在递归函数中维护一个外部变量,该变量用于记录递归信息,对该变量的回溯一般放在DFS函数的尾部,在到达递归的最深处开始回溯外部变量。

DFS与回溯代码模板:

type function(){
vector<vector<int>> res;	//用于记录结果
vector<int> path; // 用于记录递归信息
if(正常运行的条件){
	dfs(一般传入数、vector等,传入记录递归的变量,传入记录结果的变量);
	}
	return res;
}
void dfd(参数){
	if(出递归的条件) return;
	//执行的操作,对当前节点需要做的事情,非必须
	//对记录递归的变量要做的事情,path.push_back(节点信息);
	if(判断到当前节点为止,是否符合题目要求){
		符合题目要求就按要求把数据放进res中。
	}
	dfs(下一个节点);//有时需要对下一个节点进行判断,看是否满足限制要求
	path.pop_back();	//回溯,最重要的一步,在递归完成,退出当前节点的时候,需要把记录递归的变量进行回溯
}

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int target) {
        vector<vector<int>> res;
        vector<int> path;
        if(root){
            dfs(root,res,path,target);
        }
        return res;

    }
    void dfs(TreeNode* root,vector<vector<int>>& res, vector<int>& path,int target){
        if(root==nullptr) return;
        target-=root->val;
        path.push_back(root->val);
        if(target==0&&root->left==nullptr&&root->right==nullptr){
            res.push_back(path);
        }
        dfs(root->left,res,path,target);
        dfs(root->right,res,path,target);
        path.pop_back();
    }
};

你可能感兴趣的:(LeetCode,DFS,leetcode,矩阵,算法)