带你快速刷完67道剑指offer

No24、二叉树中和为某一值的路径

题目描述

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

示例1

输入

{10,5,12,4,7},22

返回值

[[10,5,7],[10,12]]

示例2

输入

{10,5,12,4,7},15

返回值

[]

1、带有回溯性质的解法

void FindPathCore(vector<vector<int>>&result, vector<int>  &temp, TreeNode* root, int sum) {
    if (root == nullptr) return;
    temp.push_back(root->val);
    if (root->left == nullptr && root->left == nullptr && sum == root->val) {
        result.push_back(temp);
    }
    else { 
        if (root->left) {
            FindPathCore(result, temp, root->left, sum-root->val);
        }
        if (root->right) {
            FindPathCore(result, temp, root->right, sum - root->val);
        }
    }
    temp.pop_back();//走到这里了，说明当前节点不满足要求，pop掉，返回其父亲节点

}
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
    vector<vector<int>>  result;
    vector<int>  temp;
    FindPathCore(result, temp, root, expectNumber);
    return result;
}

但这题是要求按照字典序返回结果的，所以最后应该是将result进行排序，优先返回那些长度较长的路径。所以最后应该再判断一下，可以用lambda表达式或者重载一个（）也可以

void FindPathCore(vector<vector<int>>&result, vector<int>  temp, TreeNode* root, int sum) {
    if (root == nullptr) return;
    temp.push_back(root->val);
    if (root->left == nullptr && root->left == nullptr && sum == root->val) {
        result.push_back(temp);
    }
    else { 
        if (root->left) {
            FindPathCore(result, temp, root->left, sum-root->val);
        }
        if (root->right) {
            FindPathCore(result, temp, root->right, sum - root->val);
        }
    }
    temp.pop_back();//走到这里了，说明当前节点不满足要求，pop掉，返回其父亲节点

}
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
    vector<vector<int>>  result;
    vector<int>  temp;
    FindPathCore(result, temp, root, expectNumber);
    sort(result.begin(),result.end(),[&](vector<int> v1,vector<int>v2){ return v1.size()>v2.size();});
    return result;
}

或者重载（）

struct compare {

    bool operator()(vector<int>& left, vector<int>& right) {
        return left.size() > right.size();
    }

};

void FindPathCore(vector<vector<int>>&result, vector<int>  temp, TreeNode* root, int sum) {
    if (root == nullptr) return;
    temp.push_back(root->val);
    if (root->left == nullptr && root->left == nullptr && sum == root->val) {
        result.push_back(temp);
    }
    else { 
        if (root->left) {
            FindPathCore(result, temp, root->left, sum-root->val);
        }
        if (root->right) {
            FindPathCore(result, temp, root->right, sum - root->val);
        }
    }
    temp.pop_back();//走到这里了，说明当前节点不满足要求，pop掉，返回其父亲节点

}
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
    vector<vector<int>>  result;
    vector<int>  temp;
    FindPathCore(result, temp, root, expectNumber);
    sort(result.begin(),result.end(),compare());
    return result;
}

二刷：

二刷也不太会，哭了，仔细想想其实也不太难，哎还是太菜了

运行时间：2ms 占用内存：484k

void FindPathCore(TreeNode*root,vector<vector<int>>&result,vector<int>temp,int sumNum){//这一这里 temp是引用方式传值，所以当前节点不符合，还要删除掉
        if(root == nullptr) return;
        temp.push_back(root->val);
        if(root->left == nullptr && root->right == nullptr &&  sumNum == root->val)
        {
            result.push_back(temp);
        }
        else{
            if(root->left)  
                FindPathCore(root->left,result,temp,sumNum-root->val);
            if(root->right)  
                FindPathCore(root->right,result,temp,sumNum-root->val);
            
        }
        temp.pop_back();//如果不是引用方式，而是值传递，这一步是可以删掉的，是引用方式就必须要pop掉

    }
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if(root == nullptr) return vector<vector<int>>();
        
        vector<vector<int>> result;
        vector<int>temp;
        FindPathCore(root,result,temp,expectNumber);
        
        sort(result.begin(),result.end(),[](const vector<int>&a,const vector<int>&b){ return a.size()>b.size();});
        return result;

    }