题目:给定二叉树的根结点指针以及链表上结点的深度,返回一个链表,代表该深度上从左往右所有结点的值。
方法
递归遍历,刚开始为深度为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;
}
};