首页 > 其他 > 详细

反转二叉树

时间:2015-06-20 18:18:51      阅读:253      评论:0      收藏:0      [点我收藏+]

Invert a binary tree.

     4
   /     2     7
 / \   / 1   3 6   9

to

     4
   /     7     2
 / \   / 9   6 3   1

Trivia:
This problem was inspired by this original tweet by Max Howell:

Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

递归版:

struct TreeNode* invertTree(struct TreeNode* root)
{
    struct TreeNode *node;

    if (root == NULL)
        return root;

    node = invertTree(root->left);
    root->left = invertTree(root->right);
    root->right = node;
    return root;
}

非递归版:

struct TreeNode* invertTree(struct TreeNode *root)
{
    struct TreeNode *node, *tmp;
    Stack treeStack;

    if (root == NULL)
        return root;

    stack_init(&treeStack, NULL);

    stack_push(&treeStack, root);
    while (treeStack.size > 0)
    {
        stack_pop(&treeStack, &node);

        tmp = node->left;
        node->left = node->right;
        node->right = tmp;

        if (node->left)
            stack_push(&treeStack, node->left);
        if (node->right)
            stack_push(&treeStack, node->right);
    }
    stack_destory(&treeStack);

    return root;
}

 

反转二叉树

原文:http://www.cnblogs.com/codelu/p/4590844.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!