程序員面試金典


程序員面試金典 -- 輸出單層結點

 

題目描述

對於一棵二叉樹,請設計一個算法,創建含有某一深度上所有結點的鏈表。

給定二叉樹的根結點指針TreeNode* root,以及鏈表上結點的深度,請返回一個鏈表ListNode,代表該深度上所有結點的值,請按樹上從左往右的順序鏈接,保證深度不超過樹的高度,樹上結點的值為非負整數且不超過100000。

 

/*
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:
void Get(TreeNode *root, int dep, ListNode* &t){
if(dep == 1){
ListNode *tmp = new ListNode(root->val);
t->next = tmp;
t = tmp;
}else{
if(root->left){
Get(root->left, dep-1, t);
}
if(root->right){
Get(root->right, dep-1, t);
}
}
}

ListNode* getTreeLevel(TreeNode* root, int dep) {
if(dep <= 0 || root == NULL ){
return NULL;
}

ListNode *ans = new ListNode(-1);
ListNode *t = ans;
Get(root, dep, t);
return ans->next;

// write code here

/*
if(root == NULL){
return NULL;
}
int tmp_int;
ListNode *ans;
ListNode *tt = ans;

queue<TreeNode *> q;
queue<int> qint;

q.push(root); qint.push(1);

while(!q.empty()){
TreeNode *tmp = q.front(); q.pop();
int tmp_int = qint.front(); qint.pop();

if(dep == tmp_int){
ListNode *t = new ListNode(tmp->val);
tt->next = t;
tt = t;
}
if(tmp->left){
q.push(tmp->left); qint.push(tmp_int + 1);
}
if(tmp->right){
q.push(tmp->right); qint.push(tmp_int + 1);
}
}
return ans->next; */

}
};

  


注意!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。



程序員面試金典 程序員面試金典 程序員面試金典 程序員面試金典 程序員面試金典 程序員面試金典 程序員面試金典2 java程序員面試金典 2 程序員面試金典 程序員面試金典
 
粤ICP备14056181号  © 2014-2021 ITdaan.com