二叉树翻转(分治思想的典型)

二叉树翻转是一道比较简单,但是又非常有意思的算法问题,当然了,对于应届生或是正在学习算法的学生来说,这道题显然是非常简单的,但对于有着多年技术经验的工程师,可能就不一定能写的出来(我指的是白板面试这类在纸类介质上进行书写的情况。)

1. 题目


话不多说,我们直接看题:

Invert a binary tree. 
     4
   /   \
  2     7
 / \   / \
1   3 6   9
to
     4
   /   \
  7     2
 / \   / \
9   6 3   1复制代码

这个问题的文字描述,简单说来就是将二叉树的左右子树全部翻转过来,使原二叉树的左子树称为新二叉树的右子树。

这个问题看起来是不是很简单?但有趣的是,有人是这样评价这个问题:

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.

产生这种问题的原因可能是多样的,不过无论是什么原因,对于一个应届生或是正在学习算法的学生来说,都是必须要忽略的,因为如果不会写算法对应的代码,我们又如何判断你是真的懂这个算法?总不能让我相信你的一面之词不是?

2. 解决代码


//这是递归方法
TreeNode* invertTree(TreeNode* root) {
    if (!root) return nullptr;
    else {
        TreeNode *temp = root->left;
        root->left = root->right;
        root->right = temp;
    }
    invertTree(root->left);
    invertTree(root->right);

    return root;
}

//进行递归简化
TreeNode* invertTree(TreeNode* root) {
    if (root) {
        invertTree(root->left);
        invertTree(root->right);
        std::swap(root->left, root->right);
    }
    return root;
}

//非递归形式@chammika
TreeNode* invertTree(TreeNode* root) {
    std::stack<TreeNode*> stk;
    stk.push(root);

    while (!stk.empty()) {
        TreeNode* p = stk.top();
        stk.pop();
        if (p) {
            stk.push(p->left);
            stk.push(p->right);
            std::swap(p->left, p->right);
        }
    }
    return root;
}复制代码

3. 解题思路


这一道题,一眼看过去,我们立刻就能得到的信息是:

  • 这是一个二叉树,与二叉树有关的算法基本都与递归相关联。
  • 这个二叉树是以链表形式实现的。
  • 翻转,将左子树变为右子树,右子树变为左子树。
  • 翻转的操作仅需改变双亲指针。

简化我们所得到的信息,即得到,我们需要通过递归来改变双亲的指针以实现翻转操作。

其实这道题算的上是一道应用分治思想的典型题了,我们按照如下步骤进行算法的编写:

  • 分解问题:二叉树的翻转涉及到左右子树,故而我们可以采用二分的方式进行分解,先进行左子树的遍历翻转,再进行右子树的遍历翻转。
  • 求解问题:当我们一次遍历到某一为nullptr的叶结点时,返回上一非空结点,交换这一非空结点的左右子树,此为递归时的出口
  • 合并问题:寻着递归链向上回溯,针对每一个分解后的问题进行合并(无相关代码,由递归性质决定)。

综上,我们可以看到,先设计一个递归出口,何时能够让递归终止if (!root) return nullptr; 此为出口的一种,我们也可以将递归出口设置在最后,此时我们需要设置的是满足什么条件才能开始递归if (root) { /* ... */ } 条件不满足时,我们可以开始递归的回溯return root

总的来说,这道题还是非常简单的(手动滑稽)。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值