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

输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
		val(x), left(NULL), right(NULL) {
	}
};
class Solution {
public:
	/*
		思路: 前序遍历求和
	*/
	vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
		vector<vector<int>> pathVectors;
		vector<int> pathVector;
		FindPath(root, expectNumber, pathVectors, pathVector);
		return pathVectors;
	}

	void FindPath(TreeNode* root, int expectNumber, vector<vector<int>>& pathVectors, vector<int>&pathVector) {
		do
		{
			if (root == NULL) {
				break;
			}
			pathVector.push_back(root->val);

			int sum = this->sum(pathVector);
			if (sum == expectNumber && root->left == NULL && root->right == NULL) {
				// 叶子节点, 且值为目标值
				pathVectors.push_back(pathVector);
			}
			FindPath(root->left, expectNumber, pathVectors, pathVector);
			FindPath(root->right, expectNumber, pathVectors, pathVector);

			// 删除最后一个
			pathVector.pop_back();

		} while (0);

		return;
	}

	int sum(vector<int>& v) {
		int s = 0;
		for (auto it : v) {
			s += it;
		}
		return s;
	}
};
测试

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_37518595/article/details/86015048