- 对于一棵二叉树,请设计一个算法,建立含有某一深度上全部结点的链表。 给定二叉树的根结点指针TreeNode* root,以及链表上结点的深度,请返回一个链表ListNode,表明该深度上全部结点的值,请遵循树上从左往右的顺序连接,保证深度不超过树的高度,树上结点的值为非负整数且不超过100000。
- 答案:这个题目的意思就是输出二叉树的某一层的全部元素,这个首先想到的是层次遍历,层次遍历最简单的方法就是用队列实现,咱们传统的层次遍历方法是能够输出全部元素,那么如何区分相邻两层之间的元素呢? 其实咱们能够用两个整数变量line1,line2来记录相邻两层的元素个数,其中line1表明出栈那一层留下的元素个数,line2表明下一层进栈元素的个数,每当line1为0的时候,说明上一层已经所有出栈,下一层已经所有入栈,那么层次遍历层数就加一,这个时候将line2的值复制给line1,line2=0,当遍历到第dep层的时候,便把那一层的全部元素输出,中止遍历。代码实现以下:
/*
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) {
// write code here
if(dep<=0||root==NULL)
return NULL;
ListNode* list=new ListNode(-1);
ListNode* listHead=list;
queue<TreeNode*> qu;
qu.push(root);
int lines1=1,lines2=0,num=1;
while(!qu.empty())
{
if(num==dep)
{
for(int i=0;i<lines1;i++)
{
TreeNode* root1=qu.front();
list->next=new ListNode(root1->val);
list=list->next;
qu.pop();
}
return listHead->next;
}
TreeNode* root1=qu.front();
if(root1->left)
{
qu.push(root1->left);
lines2++;
}
if(root1->right)
{
qu.push(root1->right);
lines2++;
}
qu.pop();
if(--lines1==0)
{
lines1=lines2;
lines2=0;
num++;
}
}
return listHead->next;
}
};