在递归函数中先搜索左节点,再搜索右节点,如果为叶子节点,则将其加入向量中,两棵树搜索完之后,按顺序和逆序对比两个向量中的元素,如果全部相等,则返回true,否则返回false。
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root1 TreeNode类
* @param root2 TreeNode类
* @return bool布尔型
*/
bool leafSimilar(TreeNode* root1, TreeNode* root2) {
vector<int> tree_1;
vector<int> tree_2;
searchTree(root1, tree_1);
searchTree(root2, tree_2);
if (tree_1.size() != tree_2.size()) {
return false;
}
for (int i = 0; i < tree_1.size(); i++) {
if (tree_1.at(i) != tree_2.at(tree_2.size() - i - 1)) {
return false;
}
}
return true;
}
void searchTree(TreeNode * node, vector<int> & tree) {
if (node->left == nullptr && node->right == nullptr) {
tree.push_back(node->val);
return;
}
if (node->left != nullptr) {
searchTree(node->left, tree);
}
if (node->right != nullptr) {
searchTree(node->right, tree);
}
}
};



京公网安备 11010502036488号