二叉树是一种更为典型的树状结构。如它名字所描述的那样,二叉树是每个节点最多有两个子树的树结构,通常子树被称作“左子树”和“右子树”。
二叉树的遍历
先(前)序遍历
前序遍历首先访问根节点,然后遍历左子树,最后遍历右子树。(根-左->右)
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
dfs(root);
return res;
}
private:
vector<int> res;
void dfs(TreeNode* root)
{
if (root == nullptr) return;
res.push_back(root->val);
dfs(root->left);
dfs(root->right);
}
};
中序遍历
中序遍历是先遍历左子树,然后访问根节点,然后遍历右子树。(左->根->右)
通常来说,对于二叉搜索树,我们可以通过中序遍历得到一个递增的有序序列。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
dfs(root);
return res;
}
private:
vector<int> res;
void dfs(TreeNode* root)
{
if (root == nullptr) return;
dfs(root->left);
res.push_back(root->val);
dfs(root->right);
}
};
后序遍历
后序遍历是先遍历左子树,然后遍历右子树,最后访问树的根节点。(左->右->根)
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
dfs(root);
return res;
}
private:
vector<int> res;
void dfs(TreeNode* root)
{
if (root == nullptr) return;
dfs(root->left);
dfs(root->right);
res.push_back(root->val);
}
};
层序遍历
层序遍历就是逐层遍历树结构。
广度优先搜索是一种广泛运用在树或图这类数据结构中,遍历或搜索的算法。 该算法从一个根节点开始,首先访问节点本身。 然后遍历它的相邻节点,其次遍历它的二级邻节点、三级邻节点,以此类推。
当我们在树中进行广度优先搜索时,我们访问的节点的顺序是按照层序遍历顺序的。
通常,我们使用一个叫做队列的数据结构来帮助我们做广度优先搜索
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
bfs(root);
return res;
}
private:
vector<vector<int>> res;
queue<TreeNode*> que;
void bfs(TreeNode* root)
{
if (root != nullptr) que.push(root);
while(!que.empty())
{
int n = que.size();
vector<int> tmp;
for (int i = 0; i < n; i++)
{
TreeNode* cur = que.front();
que.pop();
tmp.push_back(cur->val);
if (cur->left != nullptr) que.push(cur->left);
if (cur->right != nullptr) que.push(cur->right);
}
res.push_back(tmp);
}
}
};
这里要注意队列的大小要用变量保存, 因为for循环过程中队列大小发生了改变
递归求解树的问题
自顶向下
当遇到树问题时,请先思考一下两个问题:
- 你能确定一些参数,从该节点自身解决出发寻找答案吗?
- 你可以使用这些参数和节点本身的值来决定什么应该是传递给它子节点的参数吗?
如果答案都是肯定的,那么使用 “自顶向下” 的递归来解决此问题。
自底向上
对于树中的任意一个节点,如果你知道它子节点的答案,你能计算出该节点的答案吗?
如果答案是肯定的,那么 “自底向上” 的递归可能是一个不错的解决方法。
题目:二叉树的最大深度
class Solution {
public:
int maxDepth(TreeNode* root) {
dfs(root, 1);
return res;
}
private:
int res = 0;
void dfs(TreeNode* root, int depth)
{
if (!root) return;
if (!root->left && !root->right)
{
res = max(res, depth);
}
dfs(root->left, depth + 1);
dfs(root->right, depth + 1);
}
};
题目:对称二叉树
class Solution {
public:
bool isSymmetric(TreeNode* root) {
if (!root) return true;
return dfs(root->left, root->right);
}
private:
bool dfs(TreeNode* left, TreeNode* right)
{
if (!left && !right) return true;
if (!left || !right || left->val != right->val)
return false;
return dfs(left->left, right->right) && dfs(left->right, right->left);
}
};
题目:从中序与后序遍历序列构造二叉树
后序遍历的最后一个节点为根节点,从中序列表中找出根节点位置,左边为左子树,右边为右子树,接着递归即可