二叉树的所有路径

给你一个二叉树的根节点 root,按任意顺序返回所有从根节点到叶子节点的路径。

示例如下:

本题需要使用前序遍历加回溯,我们要把路径记下来然后回溯退出一个路径再进入另一个路径。

1.迭代法

前序遍历迭代来模仿遍历路径的过程,需要一个栈存放节点以及一个栈存放对应遍历的路径,实现代码如下:

//Solution1---迭代法 
vector<string> binaryTreePaths(TreeNode* root) {
	stack<TreeNode*> treeSt; //保存树的遍历节点
	stack<string> pathSt; //保存遍历路径的结点
	vector<string> result; //保存结果
	if (root == NULL) return result;
	treeSt.push(root);
	pathSt.push(to_string(root->val));
	while(!treeSt.empty()) {
		TreeNode* node = treeSt.top(); treeSt.pop();
		string path = pathSt.top(); pathSt.pop();
		if(node->left == NULL && node->right == NULL) {
			result.push_back(path);
		}
		if(node->right) {
			treeSt.push(node->right);
			pathSt.push(path+"->"+to_string(node->right->val));
		}
		if(node->left) {
			treeSt.push(node->left);
			pathSt.push(path+"->"+to_string(node->left->val));
		}
	}
	return result;
} 

2.递归法

递归三部曲:

(1)确定参数和返回值:参数为传入的结点,记录路径的path以及存放结果,无返回值;

(2)确定终止条件:当遇到根节点时处理结果;

(3)确定单层递归的逻辑:处理中间节点放入path中,然后进行递归和回溯。

实现代码如下:

//Solution2--递归法
void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
	path.pash_back(cur->val);
	if (cur->left == NULL && cur->right == NULL) { //找到叶子节点就终止 
		string spath;
		for (int i = 0; i<path.size()-1; i++) {
			sPath += to_string(path[i]);
			sPath += "->";
		}
		spath += to_string(path[path.size()-1]);
		result.push_back(spath);
		return;
	}
	if (cur->left) {
		traversal(cur->left, path, result);
		path.pop_back(); //回溯 
	}
	if(cur->right) {
		traversal(cur->right, path, result);
		path.pop_back(); //回溯 
	}
}

vector<string> binaryTreePaths(TreeNode* root) {
	vector<string> result;
	vector<int> path;
	if (root == NULL) return result;
	traversal(root, path, result);
	return result;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值