257. 二叉树的所有路径 题目描述: // 给你一个二叉树的根节点 root ,按 任意顺序 ,返回所有从根节点到叶子节点的路径。 // 叶子节点 是指没有子节点的节点。
1.基于先序遍历的解决
对于这道题很容易就可以看出来是个遍历问题的衍生,而且根据题目要求是输出路径,所以我们可以联想到使用先序遍历(根左右的顺序来进行路径打印)。 需要处理的是每个节点的路径的保存要确保不对其它的节点路径造成影响,所以使用递归的话在其回溯过程会需要处理那些多余的叶节点。 如在上述图示中,红色箭头表示深度向下打印path的路径,灰色虚线箭头表示到达叶节点之后的回溯。在我们完成A->B->D路径保存后,需要回溯到B重新进行访问,所以我们在递归过程中需要一个变量记录当前节点处的path(节点的之后节点处理不能影响这个path值),需要维护一个string的vector。为了实现回溯path不能是引用形参。所以我们的代码如下所示。
class Solution {
public:
void binaryTree(TreeNode* root,vector<string> &vs,string path){
path += to_string(root->val);
if(root->left == nullptr && root->right == nullptr){
vs.push_back(path);
return;
}
if(root->left) binaryTree(root->left,vs,path+"->");
if(root->right) binaryTree(root->right,vs,path+"->");
}
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> res;
string path;
if(root == nullptr) return res;
binaryTree(root,res,path);
return res;
}
};
2.广度优先
既然是遍历相关,我们也可以考虑使用层次遍历入手,会发现通过记录更新每个节点的path队列,就可以实现叶子结点的path输出。 // 广度优先(O(N2),O(N2)) // 该方法本质上就是维护一个与遍历节点队列相对应的遍历节点所拥有的路径队列 如对于层序为123的树,其队列改变状态分别为1–>2–>2,3–>3和“1”–>“1->2,1->3”
class Solution {
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> resPath;
if(root == nullptr) return resPath;
queue<TreeNode*> queNode;
queue<string> quePath;
queNode.push(root);
quePath.push(to_string(root->val));
while(!queNode.empty()){
TreeNode* node = queNode.front();
string path = quePath.front();
queNode.pop();
quePath.pop();
if(!node->right && !node->left){
resPath.push_back(path);
}
else{
if(node->left){
queNode.push(node->left);
quePath.push(path+"->"+to_string(node->left->val));
}
if(node->right){
queNode.push(node->right);
quePath.push(path+"->"+to_string(node->right->val));
}
}
}
return resPath;
}
};
|