贪婪的君子程序员

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

2017-08-27  本文已影响32人  贪婪的君子

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

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. 解题思路


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

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

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

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

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

上一篇下一篇

猜你喜欢

热点阅读