257. 二叉树的所有路径

题目描述

给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。
叶子节点 是指没有子节点的节点。
示例 1：

输入：root = [1,2,3,null,5]
输出：[“1->2->5”,“1->3”]

示例 2：
输入：root = [1]
输出：[“1”]

思路

这道题是一个典型的二叉树遍历题，要求写出从根结点到叶子结点的所有路径。看到树形结构的遍历这几个关键词应该会想到递归的方法。从示例一看：从1到2，从2到5，然后返回2，再返回1，再从1到3，这是完整的遍历过程。大致思路就是如此，接下来我们来看看如何实现。

代码

建树

首先我们要输入结点的个数（也就是输出的数组的大小）
然后按照层序遍历的顺序输入进一个数组，NULL就用0或者-1来代替（本题这里用的是0）。然后利用二叉树的性质进行建树，即结点左孩子的标号是本身的两倍，右孩子是本身的两倍＋1

TreeNode* buildtree(int i) {
	if (i > n || node[i] == 0) {
		return NULL;
	}
	TreeNode* root = new(TreeNode);
	root->val = node[i];
	root->left = buildtree(2 * i);
	root->right = buildtree(2 * i + 1);
	return root;
}
int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> node[i];
	}
	TreeNode*root=buildtree(1);
	return 0;
}

深度优先搜索

现在就是代码的核心部分，进行深搜
用一个ans数组来保存最终的答案，path数组保存其中一个答案，path跟着递归挨层搜索。
如下：

void dfs(TreeNode* root, string path, vector<string>& ans) {
	if (root == NULL) {
		return;
	}
	path += to_string(root->val);
	if (root->left == NULL && root->right == NULL) {
		ans.push_back(path);
	}
	else {
		path += "->";//输出格式要求
		dfs(root->left, path, ans);
		dfs(root->right, path, ans);
	}
	
}
vector<string> binaryTreePaths(TreeNode* root) {
	vector<string> ans;
	dfs(root,"",ans);
	return ans;
}

好了，这就是主要的函数介绍

源码

#include<iostream>
#include<vector>
#include<string>
using namespace std;
int n;
int node[105];
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
};
TreeNode* buildtree(int i) {
	if (i > n || node[i] == 0) {
		return NULL;
	}
	TreeNode* root = new(TreeNode);
	root->val = node[i];
	root->left = buildtree(2 * i);
	root->right = buildtree(2 * i + 1);
	return root;
}
void dfs(TreeNode* root, string path, vector<string>& ans) {
	if (root == NULL) {
		return;
	}
	path += to_string(root->val);
	if (root->left == NULL && root->right == NULL) {
		ans.push_back(path);
	}
	else {
		path += "->";//输出格式要求
		dfs(root->left, path, ans);
		dfs(root->right, path, ans);
	}
	
}
vector<string> binaryTreePaths(TreeNode* root) {
	vector<string> ans;
	dfs(root,"",ans);
	return ans;
}
int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> node[i];
	}
	TreeNode*root=buildtree(1);
	vector<string> ans = binaryTreePaths(root);
	for (int i = 0; i < ans.size(); i++) {
		cout << ans[i] << "  ";
	}
	return 0;
}

运行结果

总结

内容如上，难点主要是建树过程和dfs函数（虽然也不难）。