首页 > 其他 > 详细

LeetCode 623. Add One Row to Tree

时间:2019-08-20 10:53:30      阅读:68      评论:0      收藏:0      [点我收藏+]

给定一个数字v和k,在树的第k层插入一层v

采用递归的办法,根节点因为没有根节点,要特殊处理一下

 

class Solution {
public:
    TreeNode* helper(TreeNode* node, int v, int d, bool bLeft) {
        if(d==1){
            TreeNode* head=new TreeNode(v);
            if(bLeft)
                head->left=node;
            else
                head->right=node;
            return head;
        }
        
        if(!node)
            return NULL;
        
        node->left = helper(node->left, v, d-1, true);
        node->right = helper(node->right, v, d-1, false);
        return node;
    }
    
    TreeNode* addOneRow(TreeNode* root, int v, int d) {
        if(d==1){
            TreeNode* head=new TreeNode(v);
            head->left=root;
            return head; 
        }
        
        if(!root)
            return NULL;
        
        root->left=helper(root->left, v, d-1, true);
        root->right=helper(root->right, v, d-1, false);
        return root;
    }
};

 

LeetCode 623. Add One Row to Tree

原文:https://www.cnblogs.com/leetcoder/p/11380838.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!