leetcode刷题记录：二叉树02（思路篇）

小新00772024-02-20 3:05

参考labuladong的算法小抄：https://labuladong.online/algo/data-structure/binary-tree-part1/

复习二叉树纲领篇，二叉树解题的思维模式分两类：

1、是否可以通过遍历一遍二叉树得到答案？如果可以，用一个 traverse 函数配合外部变量来实现，这叫「遍历」的思维模式。

2、是否可以定义一个递归函数，通过子问题（子树）的答案推导出原问题的答案？如果可以，写出这个递归函数的定义，并充分利用这个函数的返回值，这叫「分解问题」的思维模式。

无论使用哪种思维模式，你都需要思考：

如果单独抽出一个二叉树节点，它需要做什么事情？需要在什么时候（前/中/后序位置）做？其他的节点不用你操心，递归函数会帮你在所有节点上执行相同的操作。

226 Invert binary tree 反转二叉树

很简单，左右子树分别翻转即可，前序遍历和后序遍历都可以。

前序遍历：先把左右子节点点交换，再分别对左右字数做同样的操作。

后序遍历：先把左右子树反转，再反转自己的左右节点。

c++ 复制代码

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) {
            return root;
        }
        TreeNode *temp = root->left;
        root->left = root->right;
        root->right = temp;
        invertTree(root->left);
        invertTree(root->right);
        return root;
    }
};

116 填充节点的右侧指针

思路：传统题目是遍历所有节点，本题是遍历两个相邻节点之间的空隙。把原二叉树看做成一个三叉树.

c++ 复制代码

class Solution {
public:
    Node* connect(Node* root) {
        if(root == NULL){
            return root;
        }
        traverse(root->left, root->right);
        return root;
    }
    void traverse(Node* n1, Node* n2) {
        if(n1 == NULL || n2 == NULL){
            return;
        }
        n1->next = n2;
        traverse(n1->left, n1->right);
        traverse(n1->right, n2->left);
        traverse(n2->left, n2->right);
    }
};

114 将二叉树展开为链表

函数作用：输入root，就会将其拉平为一条链表。

用分解的算法，在后序位置，将root的右子树接到左子树下方。

c++ 复制代码

class Solution(object):
    def flatten(self, root):
        """
        :type root: TreeNode
        :rtype: None Do not return anything, modify root in-place instead.
        """
        if not root:
            return
        self.flatten(root.left)
        self.flatten(root.right)
        temp = root.right
        root.right = root.left
        root.left = None
        p = root
        while p.right:
            p = p.right
        p.right = temp
        return root