路径之积c++(输出路径节点)

给一个二叉树和目标值,找出所有从根节点到叶子节点路径总积等于给定目标和的路径。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

class TreeNode {
public:
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode() :val(0),left(nullptr),right(nullptr){}
	TreeNode(int value) :val(value), left(nullptr), right(nullptr) {}
};
vector<vector<int>> res;
void dfs(TreeNode *a, int num,const int &target, vector<int> temp) {
	if (a == nullptr) {
		return;
	}
	num = a->val*num;
	temp.emplace_back(a->val);
	if (a->left == nullptr&&a->right == nullptr&&num == target) {
		res.emplace_back(temp);
	}

	if (a->left) {
		dfs(a->left, num,target,temp);
	}
	if (a->right) {
		dfs(a->right, num,target,temp);
	}

}
int main() {
	TreeNode *a = new TreeNode(2);
	TreeNode *b = new TreeNode(9);
	TreeNode *c = new TreeNode(3);
	TreeNode *d = new TreeNode(2);
	TreeNode *e = new TreeNode(6);
	TreeNode *f = new TreeNode(7);
	
	a->left = b;
	a->right = c;
	b->left = d;
	c->left = e;
	c->right = f;

	int target;
	cout << "请输入目标值:";
	cin >> target;
	int len = 1;
	vector<int> temp;
	dfs(a,len,target,temp);

	if (res.size() == 0) {
		cout << "[" << "]"<<endl;
	}

	for (int i = 0; i < res.size(); ++i) {
		for (int j = 0; j<res[i].size(); ++j) {
			if (i == 0&&j == 0) {
				cout << "[" << "[";
			}
			else if (j == 0) {
				cout << "[";
			}
			cout << res[i][j];
			if (j != res[i].size() - 1) {
				cout << ",";
			}
			if (i == res.size()-1&&j == res[i].size()-1) {
				cout << "]" << "]"<<endl;
			}
			else if (j == res[i].size() - 1) {
				cout << "]";
			}
		}
	}

	system("pause");
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值