题解 | #合并二叉树#
合并二叉树
https://www.nowcoder.com/practice/7298353c24cc42e3bd5f0e0bd3d1d759
代码看起来有点繁琐了,主要是结点为None的情况无法放在加法的过程中去考虑,因为这样无法让None指针接收另一个指针的节点值,所以得从none结点的父节点就要考虑none这种情况。
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param t1 TreeNode类 * @param t2 TreeNode类 * @return TreeNode类 */ void addTreeNode(TreeNode* t1, TreeNode* t2) { if (t1 && t2) t1->val += t2->val; } TreeNode* mergeTrees(TreeNode* t1, TreeNode* t2) { // write code here if (!t1 && !t2) return t1; if (!t1) t1 = new TreeNode(0); if (!t2) t2 = new TreeNode(0); vector<TreeNode*> query; query.push_back(t1); query.push_back(t2); TreeNode* tmp1, * tmp2; while (!query.empty()) { tmp1 = query[0]; tmp2 = query[1]; addTreeNode(tmp1, tmp2); query.erase(query.begin(), query.begin() + 2); if (!tmp1->left && tmp2->left) tmp1->left = new TreeNode(0); if (!tmp2->left && tmp1->left) tmp2->left = new TreeNode(0); if (!tmp1->right && tmp2->right) tmp1->right = new TreeNode(0); if (!tmp2->right && tmp1->right) tmp2->right = new TreeNode(0); if (tmp1->left && tmp2->left) { query.push_back(tmp1->left); query.push_back(tmp2->left); } if (tmp1->right && tmp2->right) { query.push_back(tmp1->right); query.push_back(tmp2->right); } } return t1; } };