代码随想录 Day15 102.二叉树的层序遍历 226.反转二叉树

me331

已于 2024-03-26 19:12:42 修改

阅读量478

点赞数 17

文章标签：数据结构算法

于 2024-03-10 22:06:21 首次发布

本文链接：https://blog.csdn.net/weixin_61057535/article/details/136592921

版权

102.二叉树的层序遍历(opens new window)

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        return result;
    }
};

注意问题：

1.对于层序遍历不熟，也是在打代码的时候需要先进行自己思考之后写一写再看，如果十多分钟还没有思路的话就看答案。在while之后紧跟着就是size在for循环中会使用到，层序遍历中需要用到for循环，是因为我们需要确定每一层的元素有多少个，确保每次都是将一行的元素输出出去，在层序遍历问题中是因为需要按层来保存元素。所以一般在对于层序遍历的时候都需要在while之后添加size，之后在跟上vector。

2.层序遍历的二维数组，只需要将一个数组放入到另一个数组中去result.push_back(vec);

3. 注意需要设置size元素，因为我们需要知道一行元素已经到了结尾的位置。

107.二叉树的层次遍历II(opens new window)

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        reverse(result.begin(), result.end()); // 在这里反转一下数组即可
        return result;

    }
};

注意一点就是对于二维数组的返回如果是reverse那么就是对于每个数组的顺序进行反转，但是对于数组内的元素并不会进行反转。

199.二叉树的右视图(opens new window)

class Solution {
public:
    vector<int> rightSideView(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<int> result;
        while (!que.empty()) {
            int size = que.size();
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (i == (size - 1)) result.push_back(node->val); // 将每一层的最后元素放入result数组中
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return result;
    }
};

注意node = 的位置以及 que.pop()是需要在for循环中，而不是for循环以外因为size每次减一就更换节点。

这个地方如果还是采用的是层序遍历的模板的话，那么就应该是让size==0的时候就放入vector

637.二叉树的层平均值(opens new window)

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<double> result;
        while (!que.empty()) {
            int size = que.size();
            double sum = 0; // 统计每一层的和
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                sum += node->val;
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(sum / size); // 将每一层均值放进结果集
        }
        return result;
    }
};

429.N叉树的层序遍历(opens new window)

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            for (int i = 0; i < size; i++) {
                Node* node = que.front();
                que.pop();
                vec.push_back(node->val);
                for (int i = 0; i < node->children.size(); i++) { // 将节点孩子加入队列
                    if (node->children[i]) que.push(node->children[i]);
                }
            }
            result.push_back(vec);
        }
        return result;

    }
};

注意n叉树节点节点问题，需要去找node中的多个children是存储在vector中的。所以说可以按照node->size()说明node中的孩子节点的个数，访问孩子节点必须使用node->children，因为children是在node定义的结构体中，并且是用vector存储所以说可以通过下标来进行访问。

515.在每个树行中找最大值(opens new window)

class Solution {
public:
    vector<int> largestValues(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<int> result;
        while (!que.empty()) {
            int size = que.size();
            int maxValue = INT_MIN; // 取每一层的最大值
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                maxValue = node->val > maxValue ? node->val : maxValue;
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(maxValue); // 把最大值放进数组
        }
        return result;
    }
};

注意 INT_MIN 表示整数中最小的数值。

116.填充每个节点的下一个右侧节点指针(opens new window)

class Solution {
public:
    Node* connect(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            // vector<int> vec;
            Node* nodePre;
            Node* node;
            for (int i = 0; i < size; i++) {
                if (i == 0) {
                    nodePre = que.front(); // 取出一层的头结点
                    que.pop();
                    node = nodePre;
                } else {
                    node = que.front();
                    que.pop();
                    nodePre->next = node; // 本层前一个节点next指向本节点
                    nodePre = nodePre->next;
                }
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            nodePre->next = NULL; // 本层最后一个节点指向NULL
        }
        return root;

    }
};

其实还是层序遍历的方式，只要是一层还未结束的时候就是链接到队列的下一个节点即可，如果是一行结束，那就让next等于NULL即可。

117.填充每个节点的下一个右侧节点指针II(opens new window)

class Solution {
public:
    Node* connect(Node* root) {
        queue<Node*> que;
        if (root != NULL) que.push(root);
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            Node* nodePre;
            Node* node;
            for (int i = 0; i < size; i++) {
                if (i == 0) {
                    nodePre = que.front(); // 取出一层的头结点
                    que.pop();
                    node = nodePre;
                } else {
                    node = que.front();
                    que.pop();
                    nodePre->next = node; // 本层前一个节点next指向本节点
                    nodePre = nodePre->next;
                }
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            nodePre->next = NULL; // 本层最后一个节点指向NULL
        }
        return root;
    }
};

上一题也可以直接用这个代码

104.二叉树的最大深度(opens new window)

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (root == NULL) return 0;
        int depth = 0;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            int size = que.size();
            depth++; // 记录深度
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
        }
        return depth;
    }
};

111.二叉树的最小深度(opens new window)

class Solution {
public:
    int minDepth(TreeNode* root) {
        if (root == NULL) return 0;
        int depth = 0;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            int size = que.size();
            depth++; // 记录最小深度
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
                if (!node->left && !node->right) { // 当左右孩子都为空的时候，说明是最低点的一层了，退出
                    return depth;
                }
            }
        }
        return depth;
    }
};

在自己写的时候就比较灵活，depth++的位置比较随意，只要是符合逻辑就可以。

226. 反转二叉树

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        swap(root->left, root->right);  // 中
        invertTree(root->left);         // 左
        invertTree(root->right);        // 右
        return root;
    }
};

注意递归的方法需要设置退出点，本题是采用的前序遍历的方式，也可以采用后续遍历的方式，但是中序的方式是比较容易出错的。因为先进行invertTree(root->left)然后就swap在这以后再invertTree(root->right)其实反转的还是左子树，所以说不能简单的使用中序遍历的方式。

如果一定要用中序遍历的方式，那就需要是

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if (root == NULL) return root;
        
        invertTree(root->left);         // 左
        swap(root->left, root->right);  // 中
        invertTree(root->left);        // 右
        return root;
    }
};

也就是在swap的上下都是对于left进行invert，但是实际上是分别对于左右子树进行了遍历。

101.对称二叉树

class Solution {
public:
    bool compare(TreeNode* left, TreeNode* right) {
        // 首先排除空节点的情况
        if (left == NULL && right != NULL) return false;
        else if (left != NULL && right == NULL) return false;
        else if (left == NULL && right == NULL) return true;
        // 排除了空节点，再排除数值不相同的情况
        else if (left->val != right->val) return false;

        // 此时就是：左右节点都不为空，且数值相同的情况
        // 此时才做递归，做下一层的判断
        bool outside = compare(left->left, right->right);   // 左子树：左、 右子树：右
        bool inside = compare(left->right, right->left);    // 左子树：右、 右子树：左
        bool isSame = outside && inside;                    // 左子树：中、 右子树：中 （逻辑处理）
        return isSame;

    }
    bool isSymmetric(TreeNode* root) {
        if (root == NULL) return true;
        return compare(root->left, root->right);
    }
};

凡是用到递归就要想到递归的三部曲就是参数返回值，退出点，核心逻辑。

采用递归的方式需要自己确认参数和返回值，返回值一定是bool，参数的话需要对于左右子树进行递归比较，因此选择了输入左右子数的节点指针。

退出点的情况需要考虑全面，分别是左右空等否以及左右的值等否的的问题。

核心逻辑就是在于如果节点的值是相等的，那么就需要进行接下来的判断。接下来的判断就是对于子树的对称查看是否是相等的。在进行判断的时候需要注意一定是采用后续的遍历方式，需要先对于左右子树进行判断以后再比较中间节点，容易理解不然也没法比。

me331

关注

17
点赞
踩
10

收藏

觉得还不错? 一键收藏
1
评论
代码随想录 Day15 102.二叉树的层序遍历 226.反转二叉树

注意问题：层序遍历的二维数组，只需要将一个数组放入到另一个数组中去result.push_back(vec);注意需要设置size元素，因为我们需要知道一行元素已经到了结尾的位置。
复制链接

扫一扫