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}
}
最低公共祖先
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。