go语言二分查找上下边界

 

查找左边界

保证返回的下脚标id，id左侧的值均比查找值小，右侧的值大于或者等于查找值。

比如a[0,1,1,3,4,5,5,5,7]

查找1，返回1；

查找2，返回3；

查找5，返回5；

查找-1，返回0；

查找8，返回9；

查找右边界

保证返回的下脚标id，id右侧的值均比查找值大，左侧的值小于或者等于查找值。

比如a[0,1,1,3,4,5,5,5,7]

查找1，返回2；

查找2，返回2；

查找5，返回7；

查找-1，返回-1；

查找8，返回8；

代码

package main
import(
"fmt"
_"time"
_"sync"
)

func left_bound(nums []int, target int) int{
    if len(nums) == 0 {
        return -1
    }
    left := 0
    right := len(nums)

    for left < right {
        mid := (left + right) >>1
        if nums[mid] == target {
            right = mid
        } else if nums[mid] < target {
            left = mid + 1
        } else if nums[mid] > target {
            right = mid
        }
    }
    return left
}

func right_bound(nums []int, target int) int{
    if len(nums) == 0 {
        return -1
    }
    left := 0
    right := len(nums)

    for left < right {
        mid := (left + right) >>1
        if nums[mid] == target {
            left = mid +1
        } else if nums[mid] < target {
            left = mid + 1
        } else if nums[mid] > target {
            right = mid
        }
    }
    return left-1
}

func main(){
    a:=[]int{1,3,5,5,7,7,8}
    fmt.Println(left_bound(a,3))
    fmt.Println(right_bound(a,5))
}