题目描述
对于一棵二叉树,请设计一个算法,创建含有某一深度上所有结点的链表。
给定二叉树的根结点指针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;
}