树02--树的子结构

原创

昕光xg 2022-08-29 09:58:18 博主文章分类：数据结构与算法 ©著作权

文章标签 树的子结构剑指offer 树结点子结构 文章分类 后端开发

©著作权归作者所有：来自51CTO博客作者昕光xg的原创作品，请联系作者获取转载授权，否则将追究法律责任

树02--树的子结构-jz17

题目概述
解析&参考答案
注意事项
说明

题目概述

算法说明
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
测试用例
{8,8,7,9,2,#,#,#,#,4,7},{8,9,2}
{8,8,7,9,3,#,#,#,#,4,7},{8,9,2}
{8,8,#,9,#,2,#,5},{8,9,#,2}
{8,8,#,9,#,2,#,5},{8,9,#,3}
{8,#,8,#,9,#,2,#,5},{8,#,9,#,2}
{8,#,8,#,9,#,2,#,5},{8,#,9,3,2}
{},{8,#,9,3,2}
{8,#,9,3,2},{}
{},{}
{1,2,3,4,5,#,#,#,#,6},{1,2,#,4,5}
{1,7,3,2,6,#,#,4,5},{2,4,5}
{1,2,3},{2,3,#}
{1,8,7,9,2},{8,9,2}
{1,2,3,4,5},{1,2,3}
true
false
true
false
true
false
false
false
false
true
true
false
true
true

解析&参考答案

解析
若树B是树A的子结构，则子结构的根节点可能为树A的任意一个节点。因此需要先遍历A中的每一个节点，再判断A中某个节点为根的子树是否包含树B(recur(A,B))。
参考答案

vim jz17.go
package main

import "fmt"

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

func HasSubtree(pRoot1 *TreeNode, pRoot2 *TreeNode) bool {
    if pRoot1 == nil || pRoot2 == nil {
        return false
    }
    return reCur(pRoot1, pRoot2) || HasSubtree(pRoot1.Left, pRoot2) || HasSubtree(pRoot1.Right, pRoot2)
}

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

func main() {
    root := &TreeNode{8, &TreeNode{8, &TreeNode{9, nil, nil}, &TreeNode{2, &TreeNode{4, nil, nil}, &TreeNode{7, nil, nil}}},
        &TreeNode{7, nil, nil}}
    subTree := &TreeNode{8, &TreeNode{9, nil, nil},
        &TreeNode{2, nil, nil}}
    ret := HasSubtree(root, subTree)
    fmt.Println(ret)
}

vim jz17_test.go
package main

import "testing"

func TestHasSubtree(t *testing.T) {
    tests := []struct {
        root   *TreeNode
        sub    *TreeNode
        result bool
    }{
        {
            &TreeNode{8, &TreeNode{8, &TreeNode{9, nil, nil}, &TreeNode{2, &TreeNode{4, nil, nil}, &TreeNode{7, nil, nil}}}, &TreeNode{7, nil, nil}},
            &TreeNode{8, &TreeNode{9, nil, nil}, &TreeNode{2, nil, nil}},
            true,
        },
        {
            &TreeNode{8, &TreeNode{8, &TreeNode{9, nil, nil}, &TreeNode{3, &TreeNode{4, nil, nil}, &TreeNode{7, nil, nil}}}, &TreeNode{7, nil, nil}},
            &TreeNode{8, &TreeNode{9, nil, nil}, &TreeNode{2, nil, nil}},
            false,
        },
        {
            nil,
            nil,
            false,
        },
    }
    for _, tt := range tests {
        if actual := HasSubtree(tt.root, tt.sub); actual != tt.result {
            t.Errorf("%v and %v got %v, expected %v", tt.root, tt.sub, HasSubtree(tt.root, tt.sub), tt.result)
        }
    }
}