LeetCode Invert Binary Tree

来源:互联网 发布:网络投资平台排名 编辑:程序博客网 时间:2024/04/28 23:41

Invert a binary tree.

    4   /   \  2     7 / \   / \1   3 6   9
to
     4   /   \  7     2 / \   / \9   6 3   1
反转二叉树,首先采用递归的方式。先遍历作左子树,当指针指向NULL时,返回,赋值给tmp。接着遍历右子树,指向NULL,返回,并将该返回值赋值给左孩子,然后将tmp赋值给右孩子。此时完成了节点1的两个左右节点的交换,然后返回上一层递归,tmp指向1节点,然后遍历右子树,重复第一步步骤,完成反转。接下来的每个节点按照如上操作。知道所有节点完成反转。

/** * 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* invertTree(TreeNode* root) {    if(root == NULL){        return root;    }    TreeNode *tmp = invertTree(root->left);    root->left = invertTree(root->right);    root->right = tmp;    return root;};
第二种方法采用非递归的形式完成,此时需要借助一个stack<TreeNode*>存储遍历过的根节点。从root先反转左右子树,root入栈,然后走向左子树,此时左子树为7,而不是原来的2。接着交换根节点为7的左右子树,7入栈。当左孩子为NULL时(此时左孩子的父节点已完两个左右子女的交换),父节点出栈,指针指向右子树。重复上述过程。这个交换过程很类似于先序遍历,不过每次根节点的两个子女要进行交换。

class Solution {public:    TreeNode* invertTree(TreeNode* root) {        stack<TreeNode*> s;        TreeNode* p = root;        TreeNode* tmp = NULL;                while (p || !s.empty()) {            while (p) {                tmp = p -> left;                p -> left = p -> right;                p -> right = tmp;                s.push(p);                p = p -> left;            }            p = s.top();            s.pop();            p = p -> right;        }        return root;    }};


0 0
原创粉丝点击