【leetcode】x 的平方根

（点击上方公众号，可快速关注）

问题描述

实现 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;
}

从算法整体框架上，跟之前二分查找完全相同。但仍有一些细微不同，这个解法与其相比有两点不一样：

1. 条件不同

   前一个例子比较mid处的值跟待查询的值是否相等，而这个问题是比较mid的平方是否等于待求根的值。然后根据比较的结果要么返回，要么将解空间缩小一半。

2. 后处理逻辑不同

   前一个例子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类型，结果肯定不会溢出了。