在递归函数中先搜索左节点,再搜索右节点,如果为叶子节点,则将其加入向量中,两棵树搜索完之后,按顺序和逆序对比两个向量中的元素,如果全部相等,则返回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); } } };