(点击上方公众号,可快速关注)
问题描述
实现 int sqrt(int x)
函数。
计算并返回 x 的平方根,其中 x 是非负整数。
由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。
示例 1:
输入: 4
输出: 2
示例 2:
输入: 8
输出: 2
说明: 8 的平方根是 2.82842…,
由于返回类型是整数,小数部分将被舍去。
问题解析
这道题大体有几类解决方式
使用已有平方根函数
这明显是“作弊”啊。
使用专业性较强的算法
比如牛顿迭代法,学过数值计算的同学对这道题会比较有优势。
使用基础算法
计算机专业学的算法和数据结构基本上都是基础算法,解决这类问题实际上就是将问题建模为基础算法可求解的形式,然后套用现有算法求解。
这道题我们即使不用专业的算法,只使用简单的二分查找算法就能解决。x
的平方根q满足:0<= q <=x。从而将问题转化为在这x+1
个数中,选择一个满足条件的数,而且这些数已经排好了序。这些特点说明二分查找适合解决该问题。
下面是解法的实现:
C++的解法
int mySqrt(int x) {
long long left = 0, right = x;
while (left <= right) {
long long mid = left + (right - left) / 2;
long long squre = mid * mid;
if (squre > x) {
right = mid - 1;
} else if (squre < x) {
left = mid + 1;
} else {
return mid;
}
}
return right;
}
从算法整体框架上,跟之前二分查找完全相同。但仍有一些细微不同,这个解法与其相比有两点不一样:
条件不同
前一个例子比较
mid
处的值跟待查询的值是否相等,而这个问题是比较mid
的平方是否等于待求根的值。然后根据比较的结果要么返回,要么将解空间缩小一半。后处理逻辑不同
前一个例子
while
循环完全执行后,元素肯定没有找到,直接返回不存在;但该例子解是一定存在的,但在while
循环后,仍然有可能没有找到解,需要对left > right
做进一步的处理,这时候right
值就是最终的解。
除此之外,算法的内核完全相同。
最后需要注意一点:int
类型的值相乘的结果使用int
保存会导致溢出,所以要使用64位的整型,这点很容易忽略。
Haskell的解法
mySort :: Integer -> Integer
mySort x = loop 0 x
where loop left right | left == right = right
| left > right = right
| mid ^ 2 == x = mid
| mid ^ 2 > x = loop left (mid-1)
| mid ^ 2 < x = loop (mid+1) right
where mid = left + (right-left) `div` 2
Haskell的代码索性使用了Integer
类型,结果肯定不会溢出了。