最低公共祖先

type Info4 struct {
    findO1  bool
    findO2  bool
    findAns *Node
}

func lowestCommonAncestor(head, o1, o2 *Node) *Node {
    if head == nil {
        return nil
    }
    return process4(head, o1, o2).findAns
}

func process4(x, o1, o2 *Node) *Info4 {
    //base case
    if x == nil {
        return &Info4{findO1: false, findO2: false, findAns: nil}
    }
    leftData := process4(x.left, o1, o2)
    rightData := process4(x.right, o1, o2)
    if leftData.findAns != nil {
        return &Info4{findO1: true, findO2: true, findAns: leftData.findAns}
    }
    if rightData.findAns != nil {
        return &Info4{findO1: true, findO2: true, findAns: rightData.findAns}
    }
    if leftData.findO1 && rightData.findO2 {
        return &Info4{findO1: true, findO2: true, findAns: x}
    }
    if leftData.findO2 && rightData.findO1 {
        return &Info4{findO1: true, findO2: true, findAns: x}
    }
    findO1 := x == o1
    findO2 := x == o2
    if leftData.findO1 || rightData.findO1 {
        if findO2 {
            return &Info4{findO1: true, findO2: true, findAns: x}
        } else {
            return &Info4{findO1: true, findO2: false, findAns: nil}
        }
    }
    if leftData.findO2 || rightData.findO2 {
        if findO1 {
            return &Info4{findO1: true, findO2: true, findAns: x}
        } else {
            return &Info4{findO1: false, findO2: true, findAns: nil}
        }
    }
    return &Info4{findO1: findO1, findO2: findO2, findAns: nil}
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。