C++:二叉树的镜像(数据结构-树)
题目描述:
操作给定的二叉树,将其变换为源二叉树的镜像。
数据范围:二叉树的节点数 0≤n≤1000 , 二叉树每个节点的值0≤val≤1000
要求: 空间复杂度 O(n) 。本题也有原地操作,即空间复杂度 O(1) 的解法,时间复杂度 O(n)
比如:
源二叉树
镜像二叉树
示例:
输入:
{8,6,10,5,7,9,11}
返回值:
{8,10,6,11,9,7,5}
解题思路:
本题考察数据结构树的使用,可用递归来解。两种解法:一种是自上而下,从根结点开始,直接交换左右子树结点,再分别对它们的左右子树进行处理,一直到最深层完成;另一种是自下而上,探索到最深层后,左右子树互换,再返回到上一层,进行左右互换,依次类推,到返回至根结点时,对根结点的左右子树进行互换,此时完成递归。
测试代码:
解法一:自上而下
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pRoot TreeNode类 * @return TreeNode类 */ TreeNode* Mirror(TreeNode* pRoot) { if(!pRoot) return pRoot; swap(pRoot->left, pRoot->right); Mirror(pRoot->left); Mirror(pRoot->right); return pRoot; } };
解法二:自下而上
/** * struct TreeNode { * int val; * struct TreeNode *left; * struct TreeNode *right; * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * }; */ class Solution { public: /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param pRoot TreeNode类 * @return TreeNode类 */ TreeNode* Mirror(TreeNode* pRoot) { if(!pRoot) return NULL; TreeNode* left = Mirror(pRoot->left); TreeNode* right = Mirror(pRoot->right); pRoot->left = right; pRoot->right = left; return pRoot; } };