给你二叉树的根节点 root
,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:[[3],[9,20],[15,7]]
思路:利用队列去存储节点,出队顺序相当于层次遍历的顺序。
解决:先让二叉树的顶点进队,出队时将其左子节点和右子节点入队(不为空),后面所有节点在出队之后都将其左右子节点入队。这里需要两个数组,一个二维数组记录所有层遍历的结果,一个一维数组记录当前层的结果。最后输出二维数组即可。注意,二叉树为空,直接返回空。
代码:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;//存储树节点的队列
if (root != NULL) que.push(root);//空二叉树直接返回空
vector<vector<int>> n;//记录结果的二维数组
while(!que.empty()){
int size=que.size();//每层节点的数量
vector<int> v;//记录每层的遍历结果
for(int i=0;i<size;i++){
TreeNode* t=que.front();//取头节点
que.pop();//出队
v.push_back(t->val);//记录出队的节点值,要保持层次遍历顺序,所以尾端插入数据
if(t->left) que.push(t->left);//当前头节点的左子节点入队
if(t->right) que.push(t->right);//当前头节点的右子节点入队
}
n.push_back(v);//将当前层的遍历结果记录
}
return n;
}
};
给你一棵二叉树的根节点 root
,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]
思路:简单来说就是交换两个子节点。
解决:这里需要先对节点遍历,再交换,采用前序遍历和后序遍历都可以。
代码:(递归法)
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;
}
};
这个题不局限于递归法,还可以用迭代法,讲解在代码随想录中代码随想录 (programmercarl.com)
给你一个二叉树的根节点 root
, 检查它是否轴对称。
示例 1:
输入:root = [1,2,2,3,4,4,3] 输出:true
思路:这里观察一下对称的要求,一个节点的左右子节点要相等;接着比较就是①左子节点的左节点和右子节点的右节点;②左子节点的右节点和右子节点的左节点。
解决:采用递归方式,递归的原则就是①和②,当然在递归之前需要判断左右子节点的其他情况。
代码:
class Solution {
public:
bool compare(TreeNode* left,TreeNode* right){
if(left==NULL&&right==NULL) return true;//都是空节点
//左右节点有一个为空的情况
else if(left!=NULL&&right==NULL) return false;
else if(left==NULL&&right!=NULL) return false;
//左右节点值不相等
else if(left->val!=right->val) return false;
bool i=compare(left->left,right->right);//比较情况①
bool j=compare(left->right,right->left);//比较情况②
return i&&j;
}
bool isSymmetric(TreeNode* root) {
if(root==NULL) return root;
return compare(root->left,root->right);
}
};