二叉树的镜像

二叉树的镜像

http://www.nowcoder.com/questionTerminal/564f4c26aa584921bc75623e48ca3011

描述

这是一篇适合初级学者的题解。这里提供2种方法。
知识点:树,递归
难度:一星


题解

题目抽象:给定一颗二叉树,将二叉树的左右孩子进行翻转,左右孩子的子树做相同的操作。

方法一:递归版本

根据题意,如果我们知道一个根节点的左孩子指针和右孩子指针,那么再改变根节点的指向即可解决问题。
也就是,需要先知道左右孩子指针,再处理根节点。显然对应遍历方式中的后序遍历。
后序遍历的模板:

void postOrder(TreeNode *root) {
    if (!root) return;
    postOrder(root->left); // left child
    postOrder(root->right); // right child
    // process root
}   

这里展示一个例子:
![图片说明](https://uploadfiles.nowcoder.com/images/20200418/284295_1587184260451_6FEC856A9DD1E83B09780CBFB41A3862 "图片标题")

代码

class Solution {
public:
    TreeNode* dfs(TreeNode *r) {
        if (!r) return nullptr;
        TreeNode *lval = dfs(r->left);
        TreeNode *rval = dfs(r->right);
        r->left = rval, r->right = lval;
        return r;
    }
    void Mirror(TreeNode *pRoot) {
        if (!pRoot) return;
        dfs(pRoot);
    }
};

时间复杂度:O(n),n为树节点的个数。每个节点只用遍历一次,所以为O(n)
空间复杂度:O(n), 每个节点都会在递归栈中存一次

方法二:非递归版本

方法一种的递归版本中遍历树的方法用的是后序遍历。所以非递归版本,只需要模拟一次树遍历。
这里模拟树的层次遍历。
层次遍历的模板为:

void bfs(TreeNode *root) {
    queue<TreeNode*> pq;
    pq.push(root);
    while (!pq.empty()) {
        int sz = pq.size();
        while (sz--) {
            TreeNode *node = pq.front(); pq.pop();
            // process node, ours tasks
            // push value to queue
            if (node->left) pq.push(node->left);
            if (node->right) pq.push(node->right);

        } // end inner while
    } // end outer while
}

所以我们的代码为;

class Solution {
public:
    void Mirror(TreeNode *pRoot) {

        queue<TreeNode*> pq;
        pq.push(pRoot);
        while (!pq.empty()) {
            int sz = pq.size();
            while (sz--) {
                TreeNode *node = pq.front(); 
                pq.pop();

                if (node->left) pq.push(node->left);
                if (node->right) pq.push(node->right);
                // our tasks
                TreeNode *cur = node->left; // 需要保存一下
                node->left = node->right;
                node->right = cur;


            } // end inner while

        } // end outer while
    }
};

时间复杂度:O(n),n为树节点的个数。每个节点只用遍历一次,所以为O(n)
空间复杂度:O(n)

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务