/*
 * function TreeNode(x) {
 *   this.val = x;
 *   this.left = null;
 *   this.right = null;
 * }
 */
/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return TreeNode类
 */
function pruneLeaves( root ) {
    // write code here
    if(root == null){
        return null;
    }
    if(root.left && root.left.left == null && root.left.right== null){
        return null;
    }
    if(root.right && root.right.left == null && root.right.right== null){
        return null;
    }
    root.left = pruneLeaves(root.left);
    root.right = pruneLeaves(root.right);
    return root;
}
module.exports = {
    pruneLeaves : pruneLeaves
};