大家好,我是大唐,刚刷完了几道经典的leetcode题,今天给大家分享一道leetcode上面的二分查找经典题型---x 的平方根,我们往下看。
题目描述
给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
示例
求解方法
这道问题要求非负整数 x 的平方根。如果遇到平方根不是整数的情况呢?只取整数部分。例如,输入 x = 8,输出为 2。8 的平方根也就是 target 值,是小数 2.82842…。2 是小于 target 的元素中、最接近 target 的元素。
因此,这道问题实际是要 查找 <= target 、且与 target 值最接近的元素。
可以用两种二分查找法来寻找:
方法一、标准二分查找法 while left <= right
这个方法应用标准二分查找法,只需改动 while 循环之后的语句 return -1为 return right。
为什么呢?我们来分析一下。当数组中没有值为 target 的元素时,因为 while 循环的条件是 left <= right,最后一次循环时搜寻区间有一个或两个元素,right = left 或 left +1。
对于本题要返回 < target、最接近 target 的元素索引的情况:
如果最后一次 while 循环时 middle > target,元素应该插入的位置是 middle - 1 ,而循环结束时 right = middle -1。
如果最后一次 while 循环时 middle < target,元素应该插入的位置是 middle。循环开始时只有 right = left = middle(如果 while 循环开始时 right = left + 1,而 middle < target,还会进入下一次循环,因此排除这种情况。),结束时 right = middle。
因此, 只需把标准二分查找代码中 while 循环之后的语句由 return -1 改为 return right 即可。
代码如下:
class Solution:def mySqrt(self, x: int) -> int:# x = 0 或 1 时,直接返回结果if x <= 1:return xleft, right = 1, xwhile left <= right:middle = left + (right - left) // 2if middle ** 2 == x:return middleelif middle ** 2 > x:right = middle - 1else:left = middle + 1return right
方法二、另一种二分查找法 while left < right
方法一,也就是标准二分查找法,是通过不断缩小搜索范围来查找某个元素。但是我们解决这一类型问题时发现,target 的取值可能是介于两个元素中间,虽然 middle 不等于 target,也许它就是最接近 target 取值的元素。比如本题例子求 8 的平方根,middle = 2 时。
对于本题要返回 < target、最接近 target 的元素索引的情况,当 middle < target 时,middle 位置有可能是我们要找的,下一次搜寻区间应该包含 middle。因此,此时 left = middle。
与方法一不同,这里的 while 循环条件为 left < right,因此循环终止时有 left = right,这样搜寻区间还有一个元素。这就是 <= target、最接近 target 的元素,最终返回它的索引( left 或 right 都可以,两者相同)。
这里有一个小细节需要注意,因为 while 循环条件为 left < right,最后一次循环时搜寻区间有两个元素,当 left = right 时循环结束。但是我们求 middle 并不是精确的平均值,而是向下取整,这导致当搜寻区间只有两个元素时,middle 始终等于 left。 这样,当 middle < target 时,left = middle,下一次循环 middle = (left + right) /2 = left ,没有更新搜寻区间,循环无法停止。怎么办呢?
代码如下:
class Solution:def mySqrt(self, x: int) -> int:# x=0或1时,直接返回结果if x <= 1:return xleft, right = 1, xwhile left < right:middle = left + (right - left) // 2 + 1if middle ** 2 == x:return middleelif middle ** 2 > x:right = middle - 1else:left = middlereturn left