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}
}
最低公共祖先
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...