原贴地址:http://blog.leanote.com/post/dawnmagnet/e25aa1655474
题目
思路分析
这个都没什么思路可以分析的,怎么获取从左到右的叶子节点,只有一种办法,就是中序遍历,在遍历的过程中判断节点是否叶子节点,存进一个数组即可。
对于两棵树,分别进行上述操作,获得两个叶子节点的数组,对比即可获得答案。
时间已经是最优的了,因为每个节点至少得访问一遍。也无需考虑空间占用率过大的问题,因为原本的遍历节点也是O(节点数目)复杂度的,所以我们这样操作不会使用很多的空间。
C++代码
class Solution {
public:
bool isLeaf(TreeNode * node) {
return node != nullptr && node->left == nullptr && node->right == nullptr;
}
void getLeafSeries(TreeNode * root, vector<int>& vec) {
if (root == nullptr) return ;
getLeafSeries(root->left, vec);
if (isLeaf(root)) vec.push_back(root->val);
getLeafSeries(root->right, vec);
}
bool leafSimilar(TreeNode* root1, TreeNode* root2) {
vector<int> v1, v2;
getLeafSeries(root1, v1);
getLeafSeries(root2, v2);
return v1 == v2;
}
};
Rust代码
use std::rc::Rc;
use std::cell::RefCell;
type TreeNodePtr = Option<Rc<RefCell<TreeNode>>>;
pub fn isLeaf(node: &TreeNodePtr) -> bool {
if let Some(node_some) = node {
let node_borrow = node_some.borrow();
node_borrow.left.is_none() && node_borrow.right.is_none()
} else {
false
}
}
pub fn getLeafSeries(root: &TreeNodePtr, vec: &mut Vec<i32>) {
if let Some(root_some) = root {
let root_borrow = root_some.borrow();
getLeafSeries(&root_borrow.left, vec);
if (isLeaf(root)) {
vec.push(root_borrow.val);
}
getLeafSeries(&root_borrow.right, vec);
}
}
impl Solution {
pub fn leaf_similar(root1: TreeNodePtr, root2: TreeNodePtr) -> bool {
let mut v1 = vec![];
let mut v2 = vec![];
getLeafSeries(&root1, &mut v1);
getLeafSeries(&root2, &mut v2);
v1 == v2
}
}