承接上文,接下来继续实现几个功能。
二叉树的深度
int dfsDeepth(TreeNode* &root)//二叉树的深度
{
if (root == nullptr)
return 0;
else
return max(dfsDeepth(root->l), dfsDeepth(root->r)) + 1;
}
int bfsDeepth(TreeNode* &root)//二叉树的深度
{
if (root == nullptr)
return 0;
queue<TreeNode *> ret;
TreeNode *p = root;
ret.push(p);
int ans = 0;//默认初始值为0
while (!ret.empty())//只要队列不为空就继续执行
{
int sz = ret.size();//得到每一层的数量
while (sz > 0)//遍历到每一层的每一个结点
{
p = ret.front();
ret.pop();
//遍历每层的每个节点是,若该节点有左右节点,则存入队列
if (p->l != nullptr)
ret.push(p->l);
if (p->r != nullptr)
ret.push(p->r);
sz--;//每遍历到每层的一个节点,长度就自减1
}
ans++;//每遍历一层执行自增1操作
}
return ans;
}
判断是否为对称二叉树
承接上文,ABDH##I##E#J##CF#K##G##肯定不是对称二叉树,这里可以任意写出一个,ABC##D##BD##C##,一个很简单的对称二叉树。
我们的搞清楚什么是对称二叉树,就是二叉树整体结构类似镜像对称一样。不仅仅要结构对称,而且对称的地方其结点值也要相同。
如图
实现要点:
1.确定递归函数的参数和返回值
因为要比较的是两个树,参数自然也是左子树节点和右子树节点
返回值是bool类型
2.确定终止条件
节点为空的情况有:
左节点为空,右节点不为空,不对称,return false
左不为空,右为空,不对称 return false
左右都为空,对称,返回true
左右节点不为空:
左右都不为空,比较节点数值,不相同就return false
3.确定单层递归的逻辑
单层递归的逻辑就是处理左右节点都不为空,且数值相同的情况。
比较二叉树外侧是否对称:传入的是左节点的左孩子,右节点的右孩子
比较内侧是否对称,传入左节点的右孩子,右节点的左孩子
如果左右都对称就返回true ,有一侧不对称就返回false
bool check(TreeNode* &j, TreeNode* &k)//最初j,k分别是最初根节点的左节点和右节点
{
queue<TreeNode *>ret;
ret.push(j); ret.push(k);//将j,k入列,进行第一次比较
while (!ret.empty())
{
//获得队列中的两个结点
j = ret.front(); ret.pop();
k = ret.front(); ret.pop();
//递归的终止条件是两个节点都为空
if (j==nullptr && k==nullptr)
continue;
//或者两个节点中有一个为空//或者两个节点的值不相等
if ((j == nullptr || k == nullptr) || (j->val != k->val))
return false;
//比较 左节点的左孩子 和 右节点的右孩子
ret.push(j->l);
ret.push(k->r);
//比较 左节点的右孩子 和 右节点的左孩子
ret.push(j->r);
ret.push(k->l);
}
return true;
}
bool isSymmetric(TreeNode* &root)//判断对称二叉树
{
return check(root->l, root->r);//比较根节点的左节点和右节点
}
反转二叉树
TreeNode* invert(TreeNode* &root)//反转二叉树
{
if (root == nullptr)
{
return nullptr;
}
else
{
//交换左右节点,非常基础的一个知识。
//新增一个变量来交换两个旧变量的值。
TreeNode *p = root->l;
root->l = root->r;
root->r = p;
//调用函数继续交换
invert(root->l);
invert(root->r);
}
return root;
}
最后,奉上运行截图