翻转二叉树-LintCode

最新推荐文章于 2019-06-12 14:13:50 发布

zhaokane

最新推荐文章于 2019-06-12 14:13:50 发布

阅读量205

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode

本文链接：https://blog.youkuaiyun.com/zhaohengchuan/article/details/77426380

LintCode 专栏收录该内容

428 篇文章

订阅专栏

本文介绍了一种翻转二叉树的算法实现，通过递归与非递归两种方式来完成二叉树节点左右子树的互换操作。递归方法简洁直观，而非递归方法则使用了栈来辅助实现。

翻转一棵二叉树
样例：
这里写图片描述
递归&非递归：

#ifndef C175_H
#define C175_H
#include<iostream>
#include<stack>
using namespace std;
class TreeNode{
public:
    int val;
    TreeNode *left, *right;
    TreeNode(int val)
    {
        this->val = val;
        this->left = this->right = NULL;
    }
};
class Solution {
public:
    /**
    * @param root: a TreeNode, the root of the binary tree
    * @return: nothing
    */
    void invertBinaryTree(TreeNode *root) {
        // write your code here
        if (root == NULL)
            return;
        if (root->left != NULL || root->right != NULL)
        {
            TreeNode *node = root->left;
            root->left = root->right;
            root->right = node;
        }
        invertBinaryTree(root->left);
        invertBinaryTree(root->right);
    }
};
class Solution2 {
public:
    /**
    * @param root: a TreeNode, the root of the binary tree
    * @return: nothing
    */
    void invertBinaryTree(TreeNode *root) {
        // write your code here
        if (root == NULL)
            return;
        stack<TreeNode*> s;
        s.push(root);
        while (!s.empty())
        {
            TreeNode *p = s.top();
            s.pop();
            TreeNode *node = p->left;
            p->left = p->right;
            p->right = node;
            if (p->left != NULL)
                s.push(p->left);
            if (p->right != NULL)
                s.push(p->right);
        }
    }
};
#endif