leecode:103. 二叉树的锯齿形层序遍历

题目描述:

给定一个二叉树,返回其节点值的锯齿形层序遍历。(即先从左往右,再从右往左进行下一层遍历,以此类推,层与层之间交替进行)。

例如:

给定二叉树 [3,9,20,null,null,15,7],

3
/
9 20
/
15 7
返回锯齿形层序遍历如下:

[
[3],
[20,9],
[15,7]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-zigzag-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解析:

层次遍历,上一层存入vector中,上一层遍历结束层次增加。

代码:


/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> res;
        queue<TreeNode *> tq;
        if(root==nullptr)
            return res;
        //遍历初始化
        tq.push(root);
        vector<int> sub;
        sub.push_back(root->val);
        res.push_back(sub);
        sub.clear();
        int i=0;
        int k=0;
        int count=0;
        while(!tq.empty()){
            //上一行遍历完成,加入res中,同时sub清零
            if(k==res[i].size())
            {   
                res.push_back(sub);
                sub.clear();
                k=0;
                i=i+1;
            }
            //左进
            if(tq.front()->left)
                {
                    tq.push(tq.front()->left);
                    sub.push_back(tq.front()->left->val);
                }
            //右进
            if(tq.front()->right)
                {
                    tq.push(tq.front()->right);
                    sub.push_back(tq.front()->right->val);
                }
            tq.pop();
            k++;
        }
        //cout<<res.size();
        for(int i=0;i<res.size();i++)
        {
            if(i%2!=0)
                reverse(res[i].begin(),res[i].end());
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我已经明白您的问题了。这是一道关于二叉树的编程题目,需要实现二叉树的建立、遍历和路径查询等功能。以下是一个简单的实现思路和界面设计: ```python # 定义二叉树结点类 class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right # 建立二叉树 def build_tree(): # TODO: 实现建树逻辑 pass # 先序遍历 def preorder_traversal(root): # TODO: 实现先序遍历逻辑 pass # 中序遍历 def inorder_traversal(root): # TODO: 实现中序遍历逻辑 pass # 后序遍历 def postorder_traversal(root): # TODO: 实现后序遍历逻辑 pass # 层序遍历 def levelorder_traversal(root): # TODO: 实现层序遍历逻辑 pass # 求根结点到指定结点的路径 def find_path(root, target): # TODO: 实现路径查找逻辑 pass # 主程序 if __name__ == '__main__': root = None while True: print("1. 建立二叉树存储结构") print("2. 求二叉树的先序遍历序列") print("3. 求二叉树的中序遍历序列") print("4. 求二叉树的后序遍历序列") print("5. 求二叉树层序遍历序列") print("6. 求根结点到指定结点的路径") print("0. 退出") choice = input("请选择0-6:") if choice == '1': root = build_tree() elif choice == '2': preorder_traversal(root) elif choice == '3': inorder_traversal(root) elif choice == '4': postorder_traversal(root) elif choice == '5': levelorder_traversal(root) elif choice == '6': target = input("请输入目标结点的值:") find_path(root, target) elif choice == '0': break else: print("输入错误,请重新选择。") ``` 您可以根据题目要求,将上述代码中的 TODO 部分补充完整,实现相应的功能。希望对您有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值