面试题27. 二叉树的镜像

面试题27. 二叉树的镜像

解题思路：大多数二叉树都可用递归实现。

递归的内部实现就是栈，所以可以用栈代替递归实现。

递归注意结束条件。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) 
    {
        if (root == NULL)
            return NULL;        
        return root=solve(root);
    }

    TreeNode *solve(TreeNode *node)
    {
        if (node == NULL)
            return node;
        TreeNode *left = solve(node->left);
        TreeNode *right = solve(node->right);
        node->left = right;
        node->right = left;

        return node;
    }
};

栈实现：队列实现也可以，因为这里不需要输出值，只需要将头节点找到即可，管它存在什么容器里，二叉树内部结构没被破坏就行。

class Solution {
public:
    TreeNode* mirrorTree(TreeNode* root) {
        stack<TreeNode*> s;
        s.push(root);
        while (!s.empty()) {
            TreeNode* node = s.top();
            s.pop();
            if (node == NULL) {
                continue;
            }
            swap(node->left, node->right);
            s.push(node->left); 
            s.push(node->right);
        }
        return root;
    }
};