题解 | #树的子结构#

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

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 * 
 * @param pRoot1 TreeNode类 
 * @param pRoot2 TreeNode类 
 * @return bool布尔型
*/
func HasSubtree( pRoot1 *TreeNode ,  pRoot2 *TreeNode ) bool {
    // write code here

    if pRoot1 == nil || pRoot2 == nil {
        return false
    }

    if pRoot1.Val == pRoot2.Val {
        leftFlag, rightFlag := true, true 
        if pRoot2.Left != nil {
            leftFlag = HasSubtree(pRoot1.Left, pRoot2.Left)
        }

        if pRoot2.Right != nil {
            rightFlag = HasSubtree(pRoot1.Right, pRoot2.Right)
        }

        if leftFlag && rightFlag {
            return true
        }
    }

    return HasSubtree(pRoot1.Left, pRoot2) || HasSubtree(pRoot1.Right, pRoot2)
}