剑指 Offer 26. 树的子结构（golang版）

• 题目描述

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:
     3
    / \
   4   5
  / \
 1   2
给定的树 B：
   4 
  /
 1
返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：
输入：A = [1,2,3], B = [3,1]
输出：false

示例 2：
输入：A = [3,4,5,1,2], B = [4,1]
输出：true

限制：
0 <= 节点个数 <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

• 思路
  1、要解决两个问题，一是在A中找到与B头结点值相等的值，二对比两棵子树是否相等
  2、注意两点，一是B树可能为空，二是A树的节点值肯能有重复的，所有要把A树中满足条件的子树都比对一遍
  3、在A树中找到与B树头结点值相等的子节点，要利用栈，遍历A树时，当节点为空或者栈为空时才退出循环，也就是，如果所有节点都遍历完成，或者放到栈中的数据全部被弹出，才遍历完成。判断遍历的节点是否为空，如果不为空，则把当前节点的指针，放入栈，并把当前节点换为下一个节点的左树的节点。如果为空，从栈中提取出上一个节点，并将上一个节点进行弹出，将从栈中提取的上一个节点的右子树的节点赋值给进行循环的节点
  4、递归两颗树，递归终止条件应该为A树为空或B树为空，B树为空时则是子树，A树为空不是子树

• 代码

func isSubStructure(A *TreeNode, B *TreeNode) bool {
	if B == nil || A == nil {
		return false
	}
	stack := make([]*TreeNode,0)
	for A != nil || len(stack) != 0 {
		if A != nil {
			if A.Val == B.Val {
				if DFS(A, B) {
					return true
				}
			}
			stack = append(stack, A)
			A = A.Left
		}else{
			A = stack[len(stack)-1]
			stack = stack[:len(stack)-1]
			A = A.Right
		}
	}
	return false
}

func DFS(A *TreeNode, B *TreeNode) bool{
	if B == nil {
		return true
	}
	if A == nil {
		return false
	}
	return (A.Val==B.Val) && DFS(A.Left, B.Left) && DFS(A.Right,B.Right)
}

• 测试代码

package main

import "fmt"

type TreeNode struct {
	Val int
	Left *TreeNode
	Right *TreeNode
}

func main() {

	var A = new(TreeNode)
	var B = new(TreeNode)
	//[4,2,3,4,5,6,7,8,9]
	//[4,8,9]
	A.Val = 3
	A.Left = &TreeNode{4,nil,nil}
	A.Right = &TreeNode{5,nil,nil}
	A.Left.Left = &TreeNode{1,nil,nil}
	A.Left.Right = &TreeNode{2,nil,nil}

	B.Val = 4
	B.Left = &TreeNode{1,nil,nil}
	res := isSubStructure(A, B)
	fmt.Println(res)
}

func isSubStructure(A *TreeNode, B *TreeNode) bool {
	if B == nil || A == nil {
		return false
	}
	stack := make([]*TreeNode,0)
	for A != nil || len(stack) != 0 {
		if A != nil {
			if A.Val == B.Val {
				if DFS(A, B) {
					return true
				}
			}
			stack = append(stack, A)
			A = A.Left
		}else{
			A = stack[len(stack)-1]
			stack = stack[:len(stack)-1]
			A = A.Right
		}
	}
	return false
}

func DFS(A *TreeNode, B *TreeNode) bool{
	if B == nil {
		return true
	}
	if A == nil {
		return false
	}
	return (A.Val==B.Val) && DFS(A.Left, B.Left) && DFS(A.Right,B.Right)
}