输出二叉树某一层的所有结点

来源:互联网 发布:淘宝旗袍模特周婷资料 编辑:程序博客网 时间:2024/05/16 19:23

题目:给定二叉树的根结点指针以及链表上结点的深度,返回一个链表,代表该深度上从左往右所有结点的值。

方法

递归遍历,刚开始为深度为dep,每往下递归一层,则深度减1,当深度为1的时候,便输出那个元素,如果先递归左子树,那么则实现从左到右打印,如果先递归右子树,则实现从右往左打印。

struct TreeNode {    int val;    struct TreeNode *left;    struct TreeNode *right;    TreeNode(int x) :            val(x), left(NULL), right(NULL) {    }};struct ListNode {    int val;    struct ListNode *next;    ListNode(int x) : val(x), next(NULL) {}};class TreeLevel {public:    ListNode *result = new ListNode(-1);    ListNode *p = result;    ListNode* getTreeLevel(TreeNode* root, int dep) {        if(root == nullptr || dep <= 0)            return nullptr;        if(dep == 1){            p->next = new ListNode(root->val);            p = p->next;        }        else{            getTreeLevel(root->left, dep-1);            getTreeLevel(root->right, dep-1);        }        return result->next;    }};
阅读全文
0 0
原创粉丝点击