代码随想录算法训练营第十四天|102. 二叉树的层序遍历、226. 翻转二叉树、101. 对称二叉树

题目链接:102. 二叉树的层序遍历

使用队列进行遍历,利用一个变量size来获取每层节点的个数。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        if(root == nullptr) return result;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()) {
            size_t size = que.size();
            vector<int> vec;
            while(size > 0) {
                size--;
                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;
    }
};

题目链接:226. 翻转二叉树

中序遍历无法得到正确结果,如果是中序遍历,先处理节点的左子树,然后节点交换左右指针,之后处理的右子树就是之前处理完的左子树。

class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        reaversal(root);
        return root;   
    }

    void reaversal(TreeNode* node) {
        if(node == nullptr) return;

        reaversal(node->left);
        reaversal(node->right);
        swap(node->right, node->left);
    }
};

题目链接:101. 对称二叉树

同时遍历两个二叉树

class Solution {
public:
    bool isSymmetric(TreeNode* root) {
        if(!root->right && !root->left) return true;
        bool result = true;
        reaversal(root->left, root->right, result);
        return result;
    }

    void reaversal(TreeNode* tree_left, TreeNode* tree_right, bool &result) {
        if(tree_left == nullptr) {
            if(tree_right == nullptr) return;
            else {result = false; return;}
        }
        if(tree_right == nullptr) {
            if(tree_left == nullptr) return;
            else {result = false; return;}
        }

        if (tree_left->val != tree_right->val) {
            result = false;
            return;
        }
        reaversal(tree_left->right, tree_right->left, result);
        reaversal(tree_left->left, tree_right->right, result);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值