给你一个非负整数 x ,计算并返回 x 的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5) 或者 x ** 0.5 。
示例 1:
输入:x = 4 输出:2
示例 2:
输入:x = 8 输出:2 解释:8 的算术平方根是 2.82842..., 由于返回类型是整数,小数部分将被舍去。
无脑直接暴力,我想了一下必超时,还是算了。
关于这种穷举的,而且步骤很少的题,要想优化就必须要进行数量级式的优化,例如O方优化到OlogN。于是暴力 < 折半 < 位运算,这里用到折半就可以过了。
根据折半的基本思路:
- while l < r:
- mid = (l + r) // 2
- if 判断语句:
- 语句1
- elif mid > x:
- 语句2
- else:
- 语句3
那么这题很明显,当 mid * mid == x时,就说明正好是mid直接返回即可。当mid * mid > x时,说明正确的值在左边,则right = mid - 1;当mid * mid < x时,说明正确的值在右边,需要left = mid + 1。要注意的是当mid * mid 大于x时,我们需要记录下mid - 1的值,也就是在不存在mid * mid > x时,返回一个大于mid且最靠近mid的数。
- def mySqrt(self, x):
- l , r = 0 , x
- res = -1
- while l<=r :
- mid = (l+r) // 2
- if mid * mid == x:
- return mid
- elif mid * mid > x:
- r = mid - 1
- res = r
- else:
- l = mid + 1
- return res
