题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
很容易想到该题是一道深度优先搜索的模板题,注意输出要求数组长度较大的数组靠前,因而最终的返回数组在返回之前需要先进行排序。注意排序的函数必须定义为static。还有一点需要注意的是,题目意为到叶子节点路径上所有整数的和为指定整数,不要错误理解成根节点到任意子节点的和为指定整数。以下是代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
static bool findmax(vector<int> a, vector<int> b){ // 比较函数定义为static类型
if(a.size()>b.size()) return true;
else return false;
}
vector<vector<int> > res;
void findroute(TreeNode* root,int expectNumber, int actualNumber,vector<int> a){
a.push_back(root->val); actualNumber+=root->val;
if(actualNumber == expectNumber&&root->left==NULL&&root->right==NULL){
res.push_back(a);
}
if(root->left!=NULL) findroute(root->left, expectNumber, actualNumber,a);
if(root->right!=NULL) findroute(root->right, expectNumber, actualNumber, a);
return;
}
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<int> a;
res.clear();
if(root!=NULL) findroute(root, expectNumber, 0, a);
sort(res.begin(),res.end(),findmax);
return res;
}
};
以上代码空间占用量较大,还有可以优化的余地。可以在private中定义一个公共的vector,并在搜索时使用回溯,return之前弹出刚刚放入的值,这样可以节省不少数组空间。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
static bool findmax(vector<int> a, vector<int> b){ // 比较函数定义为static类型
if(a.size()>b.size()) return true;
else return false;
}
vector<vector<int> > res;
vector<int> a;
void findroute(TreeNode* root,int expectNumber, int actualNumber){
a.push_back(root->val); actualNumber+=root->val;
if(actualNumber == expectNumber&&root->left==NULL&&root->right==NULL){
res.push_back(a);
}
if(root->left!=NULL) findroute(root->left, expectNumber, actualNumber);
if(root->right!=NULL) findroute(root->right, expectNumber, actualNumber);
a.pop_back();
return;
}
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
res.clear();
if(root!=NULL) findroute(root, expectNumber, 0);
sort(res.begin(),res.end(),findmax);
return res;
}
};