二叉树的层序遍历及应用(翻转二叉树)C++版

1. 非递归层序遍历

算法思想:

1. 先将根节点入队,此时队列不为空,开始循环;

2. 计算出队列的长度size,为了确保for循环一轮循环size次,也就是树的每一层的元素个数

3. 让node指针永远指向队头元素,为了暂存队头结点并将其记录保存下来,以便于执行出队操作后,依然能对出队后的结点进行操作,也就是让node指针的左右结点依次按顺序入队。

4. 再将左孩子结点和右孩子结点入队。进行下一次循环时,由于根结点已经出队,则此时que.size() = 2,for循环执行两次。此时result = [[1],[2,3]];队列结点元素分别为:【4,5,6,7】

5. 下一次循环后,que.size() 已经被重新赋值为4。现在队列里的结点都是叶子节点,所以两个if条件不成立;循环完成后,vec = [4,5,6,7],最后再把该次的vec追加到result容器中,则直接返回result =  [[1],[2,3],[4,5,6,7]]; 层序遍历结束。

代码实现如下:

void levelOrder(TreeNode* root){
        vector<vector<int>> result;
        queue<TreeNode*> que;
        if(root != nullptr) que.push(root); //先把根节点入队
        while(!que.empty()){
            int size = que.size(); //长度不是固定的,会随着队列元素个数变化
            vector<int> vec; //子容器
            for(int i = 0;i < size;i++){
                TreeNode* node = que.front(); //先把对头元素保存下来
                que.pop(); //出队访问
                vec.push_back(node->data);
                if(node->lchild) que.push(node->lchild); //分别让该节点的孩子入队
                if(node->rchild) que.push(node->rchild);
            }
            result.push_back(vec); //每结束一层,把每一层放入结果容器中,抽象成一个二维数组
        }
        for(const auto &internal:result){
            for_each(internal.begin(),internal.end(),[](int num){
                cout << num << endl;
            });
            cout << endl;
        }
    }

 

2. 层序遍历应用 (翻转二叉树)

注意:

1.层序遍历不光可以应用于翻转二叉树,这里只举出其中一个例子。

2.翻转二叉树也可以使用前序和后序遍历,但是唯独中序遍历不方便

 算法思想:

物理上的交换

1. 逻辑和层序遍历相同,代码也大致相同

2. 不同的是:在两个子结点入队之前,需要将二者进行交换,这样就实现了物理上的交换

逻辑上的交换

1. 若我们只是想在遍历的时候,是翻转后的顺序,我们可以不进行交换左右子结点,直接先让右孩子入队,左孩子后入队即可实现逻辑上的交换

代码实现如下:

//使用层序遍历翻转二叉树
    static void reverseTree(TreeNode* root){
       vector<int> result;  //结果容器
       queue<TreeNode*> que; //临时保存节点队列
       if(root != nullptr) que.push(root);
       while(!que.empty()){
          int size = que.size();
          for(int i = 0;i < size;i++){
              TreeNode* node = que.front(); //先保存对头元素
              que.pop();//对头元素出队
              swap(node->lchild,node->rchild); //让左孩子和右孩子调换
              if(node->lchild) que.push(node->lchild);
              if(node->rchild) que.push(node->rchild);
          }
       }
        for_each(result.begin(),result.end(),[](int num){
           cout << num << endl;
        });
    }

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值