代码随想录算法训练营第14、15天 | 二叉树的遍历方法 及LeetCode 226. 翻转二叉树 101. 对称二叉树

这是后面再跟上的，前几天有点事。

二叉树是一个常用的数据结构，想要使用这种数据结构首先需要掌握的就是遍历二叉树的方法。遍历二叉树的方法有常见的四种：前序遍历、中序遍历、后序遍历、层序遍历。前三种我首先掌握的是递归法，最后一种就不是递归法了。剩下的方法就等下一次再刷的时候再掌握了吧。

文章链接：

代码随想录

前序遍历

就是从根节点开始往下，从左到右。

直接用代码来看会更清晰。

void Pre_Traversal(const tree &T) {//前序遍历
	if (T) {
		cout << T->data << " ";
		Pre_Traversal(T->left);
		Pre_Traversal(T->right);
	}
}

图形可以在文章中看到。

中序遍历

中序遍历便是二叉树中的元素垂直投影到最低的位置

代码实现如下：

void Ino_Traversal(const tree &T) {//中序遍历
    if (T) {
        Ino_Traversal(T->left);
		cout << T->data << " ";
		Ino_Traversal(T->right);
	}
}

相信已经可以看出这三种遍历方法的规律了。

后序遍历

后序遍历比较难用语言来说，建议去看看资料。

代码实现如下：

void Pos_Traversal(const tree &T) {//后序遍历
	if (T) {
		Pos_Traversal(T->left);
		Pos_Traversal(T->right);
		cout << T->data << " ";
	}
}

是的，规律已经显而易见了。

那么学会这三种遍历方法后就可以去力扣上刷相应的题目了。

层序遍历

层序遍历，顾名思义，就是一层一层从左到右的进行遍历。

代码实现如下：

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();//因为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);
    }
    return result;
}

这个方法对于我来说还是可以感觉到他的巧妙的。

那么学会之后在代码随想录里有十道关于层序遍历的题，都是可以刷的。

那么还有两道题可以用学到的知识去做了。

LeetCode 226. 翻转二叉树

题目链接：

力扣

这道题可以用各种遍历方法来进行解决，只是用中序遍历来做的话会比较麻烦。

文章链接：

代码随想录

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
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;
    }
};

这道题我用的是前序遍历，理解思路后挺简单的。

LeetCode 101. 对称二叉树

题目链接：

力扣

这道题可以把一棵树当作两棵树来进行处理。这样就更好判断了。

文章链接：

代码随想录

代码实现如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
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);
    }
};

可以进行精简，不过精简后不容易理解。

今日学习6小时收获

今天步入了二叉树的领域，对我来说也是空白的领域。基本掌握了基础的知识，对我来说这种新的知识更可以让我感受到代码的魅力。最后多多重复，百炼成钢！