修剪叶子

package main
import _"fmt"
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return TreeNode类
*/
func pruneLeaves( root *TreeNode ) *TreeNode {
    if check(root){
        return nil
    }
    if root.Left!=nil{
        if check(root.Left){
            root.Left=nil
        }else{
            pruneLeaves(root.Left)
        }
    }
    if root.Right!=nil{
        if check(root.Right){
            root.Right=nil
        }else{
            pruneLeaves(root.Right)
        }
    }
    return root
}

//检查是否叶子节点的父节点
func check(root *TreeNode)bool{
    if root.Left!=nil{
        if root.Left.Left==nil&&root.Left.Right==nil{
            return true
        }
    }
    if root.Right!=nil{
        if root.Right.Left==nil&&root.Right.Right==nil{
            return true
        }
    }
    return false
}