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

原创于 2017-05-24 17:24:42 发布 · 3.9k 阅读

6 ·

CC 4.0 BY-SA版权

算法与数据结构专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种通过递归方式实现的算法，该算法可以遍历给定深度的二叉树，并将同一深度的所有节点值构建成链表。具体而言，通过递归左右子树并递减深度值来实现，最终返回一个链表，该链表包含指定深度上的所有节点值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

方法

递归遍历，刚开始为深度为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;
    }
};