题目描述
给你一个二叉树的根节点 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函数(虽然也不难)。