关闭

打印二叉树的某一层

标签: 二叉树链表
325人阅读 评论(0) 收藏 举报
分类:

题目描述

对于一棵二叉树,请设计一个算法,创建含有某一深度上所有结点的链表。
给定二叉树的根结点指针TreeNode* root,以及链表上结点的深度,请返回一个链表ListNode,代表该深度上所有结点的值,请按树上从左往右的顺序链接,保证深度不超过树的高度,树上结点的值为非负整数且不超过100000。

Code

#include <bits/stdc++.h>

using namespace std;

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* getTreeLevel(TreeNode* root, int dep) {
        ListNode* head=(ListNode* )malloc(sizeof(ListNode));
        ListNode* l=head;
        l=head;
        queue<TreeNode* >Q;
        if(root)
            Q.push(root);
        int num = 1,dept=1,tag=0;
        while(!Q.empty()&&dept<=dep) {
            int num1=0;
            for(int i=0; i<num; i++) {
                TreeNode* tmp = Q.front();
                Q.pop();
                if(dept==dep) {
                    if(!tag) {
                        head->val=tmp->val;
                        l=head;
                        tag=1;
                    } else {
                        ListNode* r=(ListNode* )malloc(sizeof(ListNode));
                        r->val=tmp->val;
                        l->next=r;
                        l=r;
                    }
                }
                if(tmp->left) {
                    Q.push(tmp->left);
                    num1++;
                }
                if(tmp->right) {
                    Q.push(tmp->right);
                    num1++;
                }
            }
            dept++;
            num=num1;
        }
        l->next=NULL;
        return head;
    }
} s;

int main() {
    TreeNode f1=TreeNode(1);
    TreeNode f2=TreeNode(2);
    TreeNode f3=TreeNode(3);
    TreeNode f4=TreeNode(4);
    f1.left=&f2;
    f2.left=&f4;
    f1.right=&f3;
    TreeNode* root = &f1;
    ListNode* ans = s.getTreeLevel(root,1);
    while(ans) {
        printf("%d -> ",ans->val);
        ans = ans->next;
    }
    puts("");
    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:282463次
    • 积分:7325
    • 等级:
    • 排名:第2984名
    • 原创:454篇
    • 转载:31篇
    • 译文:0篇
    • 评论:50条
    最新评论