C++——二叉树镜像(二叉树是否对称)

C++——二叉树镜像(二叉树是否对称)_第1张图片

本题采用递归和迭代两种实现方法。

递归方法就是通过check函数不断检测两个节点是否相等,首先传入根节点,然后递归判断子节点,结合代码很好理解。

代码如下:

bool isSymmetric(TreeNode* root) {
          return check(root,root);
      }
      bool check(TreeNode* root1,TreeNode* root2)
      {
          if(root1==nullptr&&root2==nullptr) return true;
          if(root1==nullptr||root2==nullptr) return false;
          return (root1->val==root2->val)&&check(root1->left,root2->right)
            &&check(root1->right,root2->left);

      }

迭代方法是基于二叉树的层序遍历思想。利用一个队列,若根节点不为空,则让左节点和右节点入队,记录队中的节点数目n。然后每次出队两个节点node1和node2后队中数目减少n-=2,若都为空则返回true,只有一个为空则一定不对称,返回false,都不为空则比较节点的值是否相等,若不相等,则不满足对称,返回false;若相等,则将node1的左节点、node2的右节点、node1的右节点、node2的左节点按照顺序入队,直到当前层的遍历完毕。循环上述操作直到二叉树遍历完全,队列为空。

代码如下:

bool isSymmetric(TreeNode* root){
            if(root==nullptr) return true;
            queueq;
            q.push(root->left);
            q.push(root->right);
            while(!q.empty())
            {
                int n=q.size();
                while(n>0)
                {
                    TreeNode* node1=q.front();
                    q.pop();
                    TreeNode* node2=q.front();
                    q.pop();
                    n-=2;
                    if(node1==nullptr&&node2==nullptr)
                        continue;
                    if(node1==nullptr||node2==nullptr)
                        return false;
                    if(node1->val!=node2->val)
                        return false;
                    q.push(node1->left);
                    q.push(node2->right);
                    q.push(node1->right);
                    q.push(node2->left);
                    
                }
            }
            return true;
}

 

你可能感兴趣的:(C++,对称二叉树,二叉树镜像,C++)