【剑指 offer】二叉树中和为某一值的路径

最新推荐文章于 2024-12-16 21:23:07 发布

原创最新推荐文章于 2024-12-16 21:23:07 发布 · 100 阅读

0 ·

CC 4.0 BY-SA版权

搜索同时被 3 个专栏收录

88 篇文章

订阅专栏

二叉树

50 篇文章

订阅专栏

剑指offer

45 篇文章

订阅专栏

本文介绍了一种寻找二叉树中所有路径，使得路径上节点值之和等于特定整数的算法。通过深度优先搜索(DFS)策略，递归地遍历二叉树，收集满足条件的路径，并对结果进行排序，确保路径长度较长的数组位于前列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

思路：

路径是从根节点到叶子结点的。dfs。对结果按照要求排序。

代码：

struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
        val(x), left(NULL), right(NULL) {
	}
};

bool cmp(vector<int>& A, vector<int>& B) {
    return A.size() > B.size();
}

class Solution {
public:
    vector<vector<int> > result;

    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        vector<int> temp;
        int sum = 0;
        find_(root, temp, expectNumber, sum);
        sort(result.begin(), result.end(), cmp);
        return result;
    }

    void find_(TreeNode* root, vector<int> temp, int expectNumber, int sum) {
        if (root != NULL) cout << root->val << endl;
        if (root == NULL || sum > expectNumber) return;
        temp.push_back(root->val);
        if (sum + root->val == expectNumber && root->left == NULL && root->right == NULL) {
            result.push_back(temp);
            return;
        }
        find_(root->left, temp, expectNumber, sum+root->val);
        find_(root->right, temp, expectNumber, sum+root->val);
        return;
    }
};