翻转二叉树（递归与非递归）

2023年1月5日 278次阅读 来源: 递归与分治算法

翻转一棵二叉树

样例

  1         1
 / \       / \
2   3  => 3   2
   /       \
  4         4

递归版本

先翻转左子树，后翻转右子树，然后对整个树进行翻转

void swapTree(TreeNode *&root){
    TreeNode *tmp = root->left;
    root->left = root->right;
    root->right = tmp;
}

void invertBinaryTree(TreeNode *root) {
    // write your code here
    if(root == NULL)
        return;

    invertBinaryTree(root->left);
    invertBinaryTree(root->right);

    swapTree(root);
}

非递归版本

非递归版本用栈来实现，到访问到头节点的时候，将其左子树和右子树互换即可。

void swapTree(TreeNode *&root){
    TreeNode *tmp = root->left;
    root->left = root->right;
    root->right = tmp;
}
void invertBinaryTree(TreeNode *root) {
    // write your code here
    if(root == NULL)
        return;
    stack<TreeNode*> stk;
    stk.push(root);
    while(!stk.empty())
    {
        TreeNode *tmp = stk.top();
        stk.pop();
        swapTree(tmp);
        if(tmp->left)
            stk.push(tmp->left);
        if(tmp->right)
            stk.push(tmp->right);
    }
}

(175) Invert Binary Tree
http://www.lintcode.com/zh-cn/problem/invert-binary-tree/）

    原文作者：递归与分治算法
    原文地址: https://blog.csdn.net/zwhlxl/article/details/47171083
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞