二分查找算法 golang实现

二分查找算法 golang实现

算法描述

在一组有序数组中，将数组一分为二，将要查询的元素和分割点进行比较，分为三种情况:

• 相等直接返回
• 元素大于分割点，在分割点右侧继续查找
• 元素小于分割点，在分割点左侧继续查找

时间复杂度：O(lgn)

要求

必须是有序的数组，并能支持随机访问。

代码

package main

import "fmt"

func binSearch(arr []int, findData int) int {
	low := 0
	high := len(arr) - 1
	for low <= high {
		mid := (low + high) / 2
		if arr[mid] > findData {
			high = mid - 1
		} else if arr[mid] < findData {
			low = mid + 1
		} else {
			return mid
		}
	}
	return -1
}

func main() {
	arr := make([]int, 1024 * 1024, 1024 * 1024)
	for i := 0; i < 1024 * 1024; i++ {
		arr[i] = i + 1
	}
	id := binSearch(arr, 1024)
	if id != -1 {
		fmt.Println(id, arr[id])
	}  else {
		fmt.Println("can't find data")
	}
}