Files
LeetCode-Go/leetcode/0098.Validate-Binary-Search-Tree/98. Validate Binary Search Tree.go
2020-08-07 17:06:53 +08:00

53 lines
1.2 KiB
Go
Raw Permalink Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package leetcode
import "math"
import (
"github.com/halfrost/LeetCode-Go/structures"
)
// TreeNode define
type TreeNode = structures.TreeNode
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
// 解法一,直接按照定义比较大小,比 root 节点小的都在左边,比 root 节点大的都在右边
func isValidBST(root *TreeNode) bool {
return isValidbst(root, math.Inf(-1), math.Inf(1))
}
func isValidbst(root *TreeNode, min, max float64) bool {
if root == nil {
return true
}
v := float64(root.Val)
return v < max && v > min && isValidbst(root.Left, min, v) && isValidbst(root.Right, v, max)
}
// 解法二,把 BST 按照左中右的顺序输出到数组中,如果是 BST则数组中的数字是从小到大有序的如果出现逆序就不是 BST
func isValidBST1(root *TreeNode) bool {
arr := []int{}
inOrder(root, &arr)
for i := 1; i < len(arr); i++ {
if arr[i-1] >= arr[i] {
return false
}
}
return true
}
func inOrder(root *TreeNode, arr *[]int) {
if root == nil {
return
}
inOrder(root.Left, arr)
*arr = append(*arr, root.Val)
inOrder(root.Right, arr)
}