leetcode之69x的平方根Golang

题目描述

实现 int sqrt(int x) 函数。

计算并返回 x 的平方根，其中 x 是非负整数。

由于返回类型是整数，结果只保留整数的部分，小数部分将被舍去。

示例 1:

输入: 4
输出: 2

示例 2:

输入: 8
输出: 2
说明: 8 的平方根是 2.82842..., 
     由于返回类型是整数，小数部分将被舍去。

算法

直接使用二分查找法，low=0,high=x，循环条件为low<=high

代码

func mySqrt(x int) int {
	// 采用二分法实现
	low, high := 0, x
	for low <= high {
		mid := (low + high) / 2
		mul := mid * mid
		if mul == x {
			return mid
		} else if mul < x {
			low = mid + 1
		} else {
			if (mid-1)*(mid-1) <= x {
				return mid - 1
			}
			high = mid - 1
		}
	}
	return 0
}