首页 > 学院 > 开发设计 > 正文

257. Binary Tree Paths(打印二叉树所有路径)

2019-11-14 09:13:09
字体:
来源:转载
供稿:网友
Given a binary tree, return all root-to-leaf paths.For example, given the following binary tree: 1 / /2 3 / 5All root-to-leaf paths are:["1->2->5", "1->3"]

首先我采用了dfs,但是我的方法有点麻烦。

方法一:dfs

/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public: vector<string> binaryTreePaths(TreeNode* root) { vector<string> res; if(root == NULL) return res; string str = std::to_string(root->val); if(root->left != NULL || root->right != NULL){ dfs(root->left, res, str); dfs(root->right, res, str); } else res.push_back(str); return res; } void dfs(TreeNode* root, vector<string>& res, string str){ if(root == NULL) return; str += "->" + std::to_string(root->val); if(root->left == NULL && root->right == NULL){ res.push_back(str); return ; } dfs(root->left, res, str); dfs(root->right, res, str); }};

然后我参照别人的dfs,重新用Python写了一遍,简化了一些步骤。

下面是dfs的简洁版本:

# Definition for a binary tree node.# class TreeNode(object):# def __init__(self, x):# self.val = x# self.left = None# self.right = Noneclass Solution(object): def binaryTreePaths(self, root): if not root: return [] res = [] self.dfs(root, res, "") return res; def dfs(self, root, res, ls): if not root.left and not root.right: res.append(ls+str(root.val)); return res if root.left: self.dfs(root.left, res, ls+str(root.val)+"->") if root.right: self.dfs(root.right, res, ls+str(root.val)+"->")

非递归版的dfs,类似于二叉树的前序遍历,使用stack:

class Solution(object): def binaryTreePaths(self, root): res = [] if not root: return res self.dfs(root, res) return res; def dfs(self, root, res): if not root: return [] stack = [(root, "")] while stack: node, ls = stack.pop() if not node.left and not node.right: res.append(ls+str(node.val)) if node.right: stack.append((node.right, ls+str(node.val)+"->")) if node.left: stack.append((node.left, ls+str(node.val)+"->")) return res

方法二:采用广度优先BFS,使用队列:

class Solution(object): def binaryTreePaths(self, root): res = [] if not root: return res self.bfs(root, res) return res; def bfs(self, root, res): if not root: return [] queue = collections.deque([(root, "")]) while queue: node, ls = queue.popleft() if not node.left and not node.right: res.append(ls+str(node.val)) if node.left: queue.append((node.left, ls+str(node.val)+"->")) if node.right: queue.append((node.right, ls+str(node.val)+"->")) return res
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表