版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ghscarecrow/article/details/82048290
此题出自牛客网的剑指offer。
题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
由于需要打印出二叉树中结点值的和为输入整数的所有路径,所以我们需要定义一个二维数组链表来存储每一条路径。而算法的设计我们可以采用DFS算法,当成功找到一条符合要求的路径时则将其添加到我们的结果集当中,然后再将其进行回退寻找另外符合条件的路径直到所有路径均被遍历完毕。
代码如下
Java版本:
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<Integer> list = new ArrayList<>();
ArrayList<ArrayList<Integer>> allList = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root==null){
return allList;
}
list.add(root.val);
target -= root.val;
if(root.left==null && root.right==null && target==0){
//此时已经找到一条符合要求的路径
allList.add(new ArrayList<Integer>(list));
}
FindPath(root.left,target);
FindPath(root.right,target);
//进行回退
list.remove(list.size()-1);
return allList;
}
}
C++版本
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
vector<vector<int>> allList;
vector<int> list;
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
{
return allList;
}
list.push_back(root->val);
expectNumber -= root->val;
if(root->left==NULL && root->right==NULL && expectNumber == 0)
{
allList.push_back(list);
}
FindPath(root->left,expectNumber);
FindPath(root->right,expectNumber);
list.pop_back();
return allList;
}
};